diff options
Diffstat (limited to 'tensorflow/contrib/cmake/tf_tools.cmake')
-rw-r--r-- | tensorflow/contrib/cmake/tf_tools.cmake | 52 |
1 files changed, 52 insertions, 0 deletions
diff --git a/tensorflow/contrib/cmake/tf_tools.cmake b/tensorflow/contrib/cmake/tf_tools.cmake new file mode 100644 index 0000000000..3c1fe5eac7 --- /dev/null +++ b/tensorflow/contrib/cmake/tf_tools.cmake @@ -0,0 +1,52 @@ +set(tf_tools_proto_text_src_dir "${tensorflow_source_dir}/tensorflow/tools/proto_text") + +file(GLOB tf_tools_srcs + "${tf_tools_proto_text_src_dir}/gen_proto_text_functions.cc" + "${tf_tools_proto_text_src_dir}/gen_proto_text_functions_lib.h" + "${tf_tools_proto_text_src_dir}/gen_proto_text_functions_lib.cc" +) + +set(proto_text "proto_text") + +add_executable(${proto_text} + ${tf_tools_srcs} + $<TARGET_OBJECTS:tf_core_lib> +) + +target_include_directories(${proto_text} PUBLIC + ${tensorflow_source_dir} +) + +target_link_libraries(${proto_text} PUBLIC + ${CMAKE_THREAD_LIBS_INIT} + ${PROTOBUF_LIBRARIES} + # tf_protos_cc + # re2_lib + ${jpeg_STATIC_LIBRARIES} + ${png_STATIC_LIBRARIES} + ${ZLIB_LIBRARIES} + ${jsoncpp_STATIC_LIBRARIES} + ${boringssl_STATIC_LIBRARIES} + ${CMAKE_DL_LIBS} +) + +add_dependencies(${proto_text} + tf_core_lib + # jpeg_copy_headers_to_destination + # png_copy_headers_to_destination + # re2_copy_headers_to_destination + # eigen + # tf_protos_cc + # jsoncpp + # boringssl +) + +target_compile_options(${proto_text} PRIVATE + -fno-exceptions + -DEIGEN_AVOID_STL_ARRAY +) + +# C++11 +target_compile_features(${proto_text} PRIVATE + cxx_rvalue_references +) |