diff options
| -rw-r--r-- | externals/CMakeLists.txt | 4 | ||||
| -rw-r--r-- | tests/tools/tflite_benchmark_model/CMakeLists.txt | 6 |
2 files changed, 6 insertions, 4 deletions
diff --git a/externals/CMakeLists.txt b/externals/CMakeLists.txt index b0ac9de8c..cb4073f98 100644 --- a/externals/CMakeLists.txt +++ b/externals/CMakeLists.txt @@ -4,10 +4,6 @@ if(NOT TensorFlowLite_FOUND) return() endif(NOT TensorFlowLite_FOUND) -if(BUILD_TFLITE_BENCHMARK_MODEL) - target_compile_definitions(tensorflow-lite PUBLIC "TFLITE_PROFILING_ENABLED") -endif() - if("${TARGET_OS}" STREQUAL "android") if(NOT DEFINED NDK_DIR) file(GLOB NDK_DIRS "${CMAKE_SOURCE_DIR}/tools/cross/ndk/*") 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) |