diff options
Diffstat (limited to 'applications')
-rw-r--r-- | applications/inference_process/CMakeLists.txt | 4 | ||||
-rw-r--r-- | applications/inference_process/src/inference_process.cpp | 4 |
2 files changed, 4 insertions, 4 deletions
diff --git a/applications/inference_process/CMakeLists.txt b/applications/inference_process/CMakeLists.txt index 4bd75c4..1378181 100644 --- a/applications/inference_process/CMakeLists.txt +++ b/applications/inference_process/CMakeLists.txt @@ -24,8 +24,8 @@ target_include_directories(inference_process PUBLIC include PRIVATE ${TENSORFLOW_PATH} ${TENSORFLOW_PATH}/tensorflow/lite/micro/tools/make/downloads/flatbuffers/include) target_link_libraries(inference_process PRIVATE tflu cmsis_core cmsis_device) -if (TARGET ethosu_profiler) - target_link_libraries(inference_process PRIVATE ethosu_profiler) +if (TARGET layer_by_layer_profiler) + target_link_libraries(inference_process PRIVATE layer_by_layer_profiler) endif() if (TARGET arm_profiler) target_link_libraries(inference_process PRIVATE arm_profiler) diff --git a/applications/inference_process/src/inference_process.cpp b/applications/inference_process/src/inference_process.cpp index 009706e..511a139 100644 --- a/applications/inference_process/src/inference_process.cpp +++ b/applications/inference_process/src/inference_process.cpp @@ -26,7 +26,7 @@ #include "arm_profiler.hpp" #ifdef ETHOSU -#include "ethosu_profiler.hpp" +#include "layer_by_layer_profiler.hpp" #endif #include "inference_process.hpp" @@ -202,7 +202,7 @@ bool InferenceProcess::runJob(InferenceJob &job) { // Create the TFL micro interpreter tflite::AllOpsResolver resolver; #ifdef ETHOSU - tflite::EthosUProfiler profiler; + tflite::LayerByLayerProfiler profiler; #else tflite::ArmProfiler profiler; #endif |