blob: 1887d7cbf0f1fb3c365fe5af6b9f3579f24c5531 (
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
|
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/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)
install(TARGETS tflite_run DESTINATION bin)
# TEST BUILD
nnfw_find_package(GTest)
if(NOT GTest_FOUND)
return()
endif(NOT GTest_FOUND)
## Add test cpp file
add_executable(tflite_test src/tflite_test.cc)
## Link test executable against gtest & gtest_main
target_link_libraries(tflite_test gtest gtest_main ${LIB_PTHREAD})
## install test binary for packaging
install(TARGETS tflite_test DESTINATION unittest)
|