nihui
/
ncnn
Not watched
1
0
Fork 0
Code
Releases 46 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: 86f4264c7c
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '86f4264c7c'
${ noResults }
ncnn/tools/tensorflow/CMakeLists.txt

23 lines
555 B
Raw Blame History 

  1. 

  2. find_package(Protobuf REQUIRED)

  3. 

  4. include_directories(${PROTOBUF_INCLUDE_DIR})

  5. 

  6. include_directories(${CMAKE_CURRENT_BINARY_DIR})

  7. protobuf_generate_cpp(TENSORFLOW_PROTO_SRCS TENSORFLOW_PROTO_HDRS

  8.     attr_value.proto

  9.     function.proto

  10.     graph.proto

  11.     node_def.proto

  12.     op_def.proto

  13.     resource_handle.proto

  14.     tensor.proto

  15.     tensor_shape.proto

  16.     types.proto

  17.     versions.proto

  18. )

  19. 

  20. add_executable(tensorflow2ncnn tensorflow2ncnn.cpp ${TENSORFLOW_PROTO_SRCS} ${TENSORFLOW_PROTO_HDRS})

  21. 

  22. target_link_libraries(tensorflow2ncnn ${PROTOBUF_LIBRARIES})