aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/cmake/tf_tools.cmake
blob: 5022dfee0d61321c46ffb83fd079455a95b54ccb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
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
    ${gif_STATIC_LIBRARIES}
    ${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
)