Skip to content

Commit

Permalink
Tensorflow serializer (#1114)
Browse files Browse the repository at this point in the history
* Fix convert tool compiler error

Co-authored-by: bzhang <[email protected]>
  • Loading branch information
bzhang5 and bzhang authored Sep 3, 2021
1 parent 945a371 commit 7bf86d1
Showing 1 changed file with 1 addition and 15 deletions.
16 changes: 1 addition & 15 deletions tools/convert_tool/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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")

Expand Down

0 comments on commit 7bf86d1

Please sign in to comment.