diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/tools/tflite_benchmark_model/CMakeLists.txt | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tests/tools/tflite_benchmark_model/CMakeLists.txt b/tests/tools/tflite_benchmark_model/CMakeLists.txt index c48f658c1..7e0e9d861 100644 --- a/tests/tools/tflite_benchmark_model/CMakeLists.txt +++ b/tests/tools/tflite_benchmark_model/CMakeLists.txt @@ -2,6 +2,12 @@ if (NOT BUILD_TFLITE_BENCHMARK_MODEL) return() endif(NOT BUILD_TFLITE_BENCHMARK_MODEL) +nnfw_find_package(TensorFlowLite REQUIRED) + +# TODO Remove this target_compile_definitions command, and just check its presence. +# This change is prerequisites on pre-built tensorflow-lite package support +target_compile_definitions(tensorflow-lite PUBLIC "TFLITE_PROFILING_ENABLED") + file(GLOB_RECURSE SOURCES "*.cc") nnfw_find_package(TensorFlowSource REQUIRED) |