|
|
|
@ -132,11 +132,6 @@ cc_test(version_test SRCS version_test.cc DEPS version)
|
|
|
|
|
cc_library(proto_desc SRCS var_desc.cc op_desc.cc block_desc.cc program_desc.cc DEPS shape_inference op_info operator glog version)
|
|
|
|
|
|
|
|
|
|
cc_library(op_registry SRCS op_registry.cc DEPS op_proto_maker op_info operator glog proto_desc memory_optimize_helper)
|
|
|
|
|
if(WITH_NGRAPH)
|
|
|
|
|
cc_library(ngraph_bridge SRCS ngraph_bridge.cc DEPS operator framework_proto ngraph)
|
|
|
|
|
cc_library(ngraph_operator SRCS ngraph_operator.cc DEPS ngraph_bridge operator op_info device_context tensor scope glog
|
|
|
|
|
shape_inference data_transform lod_tensor profiler)
|
|
|
|
|
endif(WITH_NGRAPH)
|
|
|
|
|
|
|
|
|
|
nv_test(op_registry_test SRCS op_registry_test.cc DEPS op_registry)
|
|
|
|
|
|
|
|
|
|