diff options
Diffstat (limited to 'tests/tools/tflite_run/CMakeLists.txt')
-rw-r--r-- | tests/tools/tflite_run/CMakeLists.txt | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/tests/tools/tflite_run/CMakeLists.txt b/tests/tools/tflite_run/CMakeLists.txt index 49d87318f..1887d7cbf 100644 --- a/tests/tools/tflite_run/CMakeLists.txt +++ b/tests/tools/tflite_run/CMakeLists.txt @@ -1,11 +1,18 @@ +if(NOT BUILD_TFLITE_RUN) + return() +endif(NOT BUILD_TFLITE_RUN) + list(APPEND TFLITE_RUN_SRCS "src/tflite_run.cc") -list(APPEND TFLITE_RUN_SRCS "src/bin_image.cc") list(APPEND TFLITE_RUN_SRCS "src/args.cc") list(APPEND TFLITE_RUN_SRCS "src/tensor_dumper.cc") list(APPEND TFLITE_RUN_SRCS "src/tensor_loader.cc") +nnfw_find_package(Boost REQUIRED) + add_executable(tflite_run ${TFLITE_RUN_SRCS}) target_include_directories(tflite_run PRIVATE src) +target_include_directories(tflite_run PRIVATE ${Boost_INCLUDE_DIRS}) + target_link_libraries(tflite_run tensorflow-lite ${LIB_PTHREAD} dl nnfw_lib_tflite) target_link_libraries(tflite_run boost_program_options boost_system boost_filesystem) |