From 7bf86d16f22b35ca798a93a9bc6ed200aa541722 Mon Sep 17 00:00:00 2001 From: bzhang5 <58622517+bzhang5@users.noreply.github.com> Date: Fri, 3 Sep 2021 10:18:57 +0800 Subject: [PATCH] Tensorflow serializer (#1114) * Fix convert tool compiler error Co-authored-by: bzhang --- tools/convert_tool/CMakeLists.txt | 16 +--------------- 1 file changed, 1 insertion(+), 15 deletions(-) diff --git a/tools/convert_tool/CMakeLists.txt b/tools/convert_tool/CMakeLists.txt index ab028141b..3a82d9d23 100644 --- a/tools/convert_tool/CMakeLists.txt +++ b/tools/convert_tool/CMakeLists.txt @@ -40,6 +40,7 @@ file(GLOB_RECURSE TF_SERIALIZER_SRCS "${CMAKE_CURRENT_SOURCE_DIR}/tensorflow/*.c list(APPEND TENGINE_LIB_SRCS ${serializer_src}) + # the generated pb.cc set(TF_PROTO_SRC ${CMAKE_CURRENT_BINARY_DIR}/tensorflow/graph.pb.cc ${CMAKE_CURRENT_BINARY_DIR}/tensorflow/function.pb.cc @@ -76,21 +77,6 @@ include_directories(${TF_PROTO_OUT_PATH}) list(APPEND TF_SERIALIZER_SRCS ${TF_PROTO_SRC}) -# the generated pb.cc -protobuf_generate_cpp(TF_PROTO_SRCS TF_PROTO_HDRS - tensorflow/graph.proto - tensorflow/function.proto - tensorflow/node_def.proto - tensorflow/op_def.proto - tensorflow/attr_value.proto - tensorflow/tensor.proto - tensorflow/tensor_shape.proto - tensorflow/types.proto - tensorflow/versions.proto - tensorflow/resource_handle.proto -) -list(APPEND TF_SERIALIZER_SRCS ${TF_PROTO_HDRS} ${TF_PROTO_SRCS}) - # MXNET file(GLOB_RECURSE MXNET_SERIALIZER_SRCS "${CMAKE_CURRENT_SOURCE_DIR}/mxnet/*.cpp")