Browse Source
modified: ge/CMakeLists.txt
modified: ge/ge_inference.mk modified: ge/ge_runner.mktags/v1.2.0
zhaoxinxin
4 years ago
3 changed files with 4 additions and 0 deletions
Unified View
Diff Options
-
+2 -0ge/CMakeLists.txt
-
+1 -0ge/ge_inference.mk
-
+1 -0ge/ge_runner.mk
+ 2
- 0
ge/CMakeLists.txt
View File
| @@ -202,6 +202,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" | ||||
| @@ -488,6 +489,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" | ||||
+ 1
- 0
ge/ge_inference.mk
View File
| @@ -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 \ | ||||
+ 1
- 0
ge/ge_runner.mk
View File
| @@ -171,6 +171,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 \ | ||||