diff options
Diffstat (limited to 'infra/nncc/cmake/packages/TensorFlowLite-1.12/Lite/CMakeLists.txt')
-rw-r--r-- | infra/nncc/cmake/packages/TensorFlowLite-1.12/Lite/CMakeLists.txt | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/infra/nncc/cmake/packages/TensorFlowLite-1.12/Lite/CMakeLists.txt b/infra/nncc/cmake/packages/TensorFlowLite-1.12/Lite/CMakeLists.txt new file mode 100644 index 000000000..068022fcf --- /dev/null +++ b/infra/nncc/cmake/packages/TensorFlowLite-1.12/Lite/CMakeLists.txt @@ -0,0 +1,41 @@ +# NOTE The followings SHOULD be defined before using this CMakeLists.txt +# +# 'TensorFlowSource_DIR' variable +# 'FlatBuffersSource_DIR' variable +# 'eigen' target +# 'gemmlowp' target +# 'neon2sse' target +# 'farmhash' target +# 'abseil' target +# +message(STATUS "Build TensorFlow Lite from ${TensorFlowSource_DIR}") + +set(TensorFlowLiteSource_DIR ${TensorFlowSource_DIR}/tensorflow/contrib/lite) + +file(GLOB CORE_SRCS "${TensorFlowLiteSource_DIR}/*.c" "${TensorFlowLiteSource_DIR}/*.cc" "${TensorFlowLiteSource_DIR}/c/*.c" "${TensorFlowLiteSource_DIR}/core/api/*.cc") +file(GLOB_RECURSE CORE_TESTS "${TensorFlowLiteSource_DIR}/*test*.cc") +list(REMOVE_ITEM CORE_SRCS ${CORE_TESTS}) + +file(GLOB_RECURSE KERNEL_SRCS "${TensorFlowLiteSource_DIR}/kernels/*.cc") +file(GLOB_RECURSE KERNEL_TESTS "${TensorFlowLiteSource_DIR}/kernels/*test*.cc") +list(REMOVE_ITEM KERNEL_SRCS ${KERNEL_TESTS}) +# Exclude buggy kernel(s) from the build +#list(REMOVE_ITEM KERNEL_SRCS "${TensorFlowLiteSource_DIR}/kernels/internal/spectrogram.cc") + +list(APPEND SRCS ${CORE_SRCS}) +list(APPEND SRCS ${KERNEL_SRCS}) + +include(CheckCXXCompilerFlag) + +CHECK_CXX_COMPILER_FLAG(-Wno-extern-c-compat COMPILER_SUPPORT_EXTERN_C_COMPAT_WARNING) + +add_library(tensorflowlite-1.12 ${SRCS}) +set_target_properties(tensorflowlite-1.12 PROPERTIES POSITION_INDEPENDENT_CODE ON) +target_include_directories(tensorflowlite-1.12 PUBLIC ${TensorFlowSource_DIR}) +target_include_directories(tensorflowlite-1.12 PUBLIC ${FlatBuffersSource_DIR}/include) +target_compile_options(tensorflowlite-1.12 PUBLIC -Wno-ignored-attributes) +if(COMPILER_SUPPORT_EXTERN_C_COMPAT_WARNING) + target_compile_options(tensorflowlite-1.12 PUBLIC -Wno-extern-c-compat) +endif(COMPILER_SUPPORT_EXTERN_C_COMPAT_WARNING) +target_compile_definitions(tensorflowlite-1.12 PUBLIC "GEMMLOWP_ALLOW_SLOW_SCALAR_FALLBACK") +target_link_libraries(tensorflowlite-1.12 eigen gemmlowp neon2sse farmhash abseil dl) |