diff options
Diffstat (limited to 'lib/layer_by_layer_profiler')
-rw-r--r-- | lib/layer_by_layer_profiler/include/layer_by_layer_profiler.hpp | 2 | ||||
-rw-r--r-- | lib/layer_by_layer_profiler/src/layer_by_layer_profiler.cpp | 4 |
2 files changed, 3 insertions, 3 deletions
diff --git a/lib/layer_by_layer_profiler/include/layer_by_layer_profiler.hpp b/lib/layer_by_layer_profiler/include/layer_by_layer_profiler.hpp index 649cfb5..a76ac41 100644 --- a/lib/layer_by_layer_profiler/include/layer_by_layer_profiler.hpp +++ b/lib/layer_by_layer_profiler/include/layer_by_layer_profiler.hpp @@ -20,7 +20,7 @@ #define LAYER_BY_LAYER_PROFILER_H #include "EventRecorder.h" -#include "tensorflow/lite/kernels/internal/compatibility.h" +#include "tensorflow/lite/micro/compatibility.h" #include "tensorflow/lite/micro/micro_profiler_interface.h" #include <memory> #include <vector> diff --git a/lib/layer_by_layer_profiler/src/layer_by_layer_profiler.cpp b/lib/layer_by_layer_profiler/src/layer_by_layer_profiler.cpp index 4f525ee..927587f 100644 --- a/lib/layer_by_layer_profiler/src/layer_by_layer_profiler.cpp +++ b/lib/layer_by_layer_profiler/src/layer_by_layer_profiler.cpp @@ -17,7 +17,7 @@ */ #include "tensorflow/lite/kernels/internal/compatibility.h" -#include "tensorflow/lite/micro/micro_error_reporter.h" +#include "tensorflow/lite/micro/micro_log.h" #include "tensorflow/lite/micro/micro_profiler.h" #include "tensorflow/lite/micro/micro_time.h" @@ -41,7 +41,7 @@ LayerByLayerProfiler::LayerByLayerProfiler(size_t max_events, Backend _backend, // NOTE: THIS PROFILER ONLY WORKS ON SYSTEMS WITH 1 NPU uint32_t LayerByLayerProfiler::BeginEvent(const char *tag) { if (num_events_ == max_events_) { - tflite::GetMicroErrorReporter()->Report("Profiling event overflow, max: %u events", max_events_); + MicroPrintf("Profiling event overflow, max: %u events", max_events_); num_events_ = 0; } |