blob: dd54dc5b539ac7f6a8d93b3ea69a747b6db2cc44 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
|
file(GLOB_RECURSE SOURCES "*.cc")
nnfw_find_package(TensorFlowSource REQUIRED)
set(TENSORFLOW_LITE_BASE "${TensorFlowSource_DIR}/tensorflow/contrib/lite")
list(APPEND SOURCES "${TENSORFLOW_LITE_BASE}/tools/benchmark/benchmark_main.cc"
"${TENSORFLOW_LITE_BASE}/tools/benchmark/benchmark_model.cc"
"${TENSORFLOW_LITE_BASE}/tools/benchmark/benchmark_params.cc"
"${TENSORFLOW_LITE_BASE}/tools/benchmark/command_line_flags.cc")
add_executable(tflite_benchmark_model ${SOURCES})
target_compile_definitions(tflite_benchmark_model PUBLIC "TFLITE_PROFILING_ENABLED")
target_link_libraries(tflite_benchmark_model tensorflow-lite ${LIB_PTHREAD} dl nnfw_util nnfw_support_tflite)
install(TARGETS tflite_benchmark_model DESTINATION bin)
|
