From: @hugo1 Reviewed-by: @sheng-nan,@xchu42 Signed-off-by: @xchu42tags/v1.2.0
@@ -201,6 +201,7 @@ set(TRAIN_SRC_LIST | |||||
"host_kernels/sub_kernel.cc" | "host_kernels/sub_kernel.cc" | ||||
"host_kernels/transdata_kernel.cc" | "host_kernels/transdata_kernel.cc" | ||||
"host_kernels/unpack_kernel.cc" | "host_kernels/unpack_kernel.cc" | ||||
"host_kernels/reformat_kernel.cc" | |||||
"graph/passes/folding_pass.cc" | "graph/passes/folding_pass.cc" | ||||
"graph/passes/get_original_format_pass.cc" | "graph/passes/get_original_format_pass.cc" | ||||
"graph/passes/guarantee_const_pass.cc" | "graph/passes/guarantee_const_pass.cc" | ||||
@@ -487,6 +488,7 @@ set(INFER_SRC_LIST | |||||
"host_kernels/slice_d_kernel.cc" | "host_kernels/slice_d_kernel.cc" | ||||
"host_kernels/dynamic_stitch_kernel.cc" | "host_kernels/dynamic_stitch_kernel.cc" | ||||
"host_kernels/identity_kernel.cc" | "host_kernels/identity_kernel.cc" | ||||
"host_kernels/reformat_kernel.cc" | |||||
"graph/passes/stop_gradient_pass.cc" | "graph/passes/stop_gradient_pass.cc" | ||||
"graph/passes/prevent_gradient_pass.cc" | "graph/passes/prevent_gradient_pass.cc" | ||||
"graph/passes/identity_pass.cc" | "graph/passes/identity_pass.cc" | ||||
@@ -164,6 +164,7 @@ OMG_HOST_SRC_FILES := \ | |||||
host_kernels/slice_d_kernel.cc \ | host_kernels/slice_d_kernel.cc \ | ||||
host_kernels/dynamic_stitch_kernel.cc \ | host_kernels/dynamic_stitch_kernel.cc \ | ||||
host_kernels/identity_kernel.cc \ | host_kernels/identity_kernel.cc \ | ||||
host_kernels/reformat_kernel.cc \ | |||||
graph/passes/stop_gradient_pass.cc \ | graph/passes/stop_gradient_pass.cc \ | ||||
graph/passes/prevent_gradient_pass.cc \ | graph/passes/prevent_gradient_pass.cc \ | ||||
graph/passes/identity_pass.cc \ | graph/passes/identity_pass.cc \ | ||||
@@ -170,6 +170,7 @@ LIBGE_LOCAL_SRC_FILES := \ | |||||
host_kernels/sub_kernel.cc \ | host_kernels/sub_kernel.cc \ | ||||
host_kernels/transdata_kernel.cc \ | host_kernels/transdata_kernel.cc \ | ||||
host_kernels/unpack_kernel.cc \ | host_kernels/unpack_kernel.cc \ | ||||
host_kernels/reformat_kernel.cc \ | |||||
graph/passes/folding_pass.cc \ | graph/passes/folding_pass.cc \ | ||||
graph/passes/get_original_format_pass.cc \ | graph/passes/get_original_format_pass.cc \ | ||||
graph/passes/guarantee_const_pass.cc \ | graph/passes/guarantee_const_pass.cc \ | ||||