|
|
|
@ -128,7 +128,7 @@ 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)
|
|
|
|
|
cc_library(op_registry SRCS op_registry.cc DEPS op_proto_maker op_info operator glog proto_desc memory_optimize_helper)
|
|
|
|
|
nv_test(op_registry_test SRCS op_registry_test.cc DEPS op_registry)
|
|
|
|
|
|
|
|
|
|
py_proto_compile(framework_py_proto SRCS framework.proto data_feed.proto)
|
|
|
|
|