diff options
Diffstat (limited to 'lib')
| -rw-r--r-- | lib/arm_profiler/include/arm_profiler.hpp | 7 | ||||
| -rw-r--r-- | lib/layer_by_layer_profiler/include/layer_by_layer_profiler.hpp | 5 |
2 files changed, 6 insertions, 6 deletions
diff --git a/lib/arm_profiler/include/arm_profiler.hpp b/lib/arm_profiler/include/arm_profiler.hpp index 0e6784d..3a487bd 100644 --- a/lib/arm_profiler/include/arm_profiler.hpp +++ b/lib/arm_profiler/include/arm_profiler.hpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2021-2022 Arm Limited. All rights reserved. + * SPDX-FileCopyrightText: Copyright 2021-2022 Arm Limited and/or its affiliates <open-source-office@arm.com> * * SPDX-License-Identifier: Apache-2.0 * @@ -19,13 +19,12 @@ #ifndef ARM_PROFILER_H #define ARM_PROFILER_H -#include "tensorflow/lite/kernels/internal/compatibility.h" -#include "tensorflow/lite/micro/micro_profiler.h" +#include "tensorflow/lite/micro/micro_profiler_interface.h" #include <memory> // TODO: Merge this profiler with EthosUprofiler. namespace tflite { -class ArmProfiler : public MicroProfiler { +class ArmProfiler : public MicroProfilerInterface { public: ArmProfiler(size_t max_events = 200); uint32_t BeginEvent(const char *tag); 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 0c50bc8..649cfb5 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 @@ -1,5 +1,5 @@ /* - * Copyright (c) 2021-2022 Arm Limited. All rights reserved. + * SPDX-FileCopyrightText: Copyright 2021-2022 Arm Limited and/or its affiliates <open-source-office@arm.com> * * SPDX-License-Identifier: Apache-2.0 * @@ -21,13 +21,14 @@ #include "EventRecorder.h" #include "tensorflow/lite/kernels/internal/compatibility.h" +#include "tensorflow/lite/micro/micro_profiler_interface.h" #include <memory> #include <vector> // NOTE: This profiler only works on systems with 1 NPU due to the use of // ethosu_reserve_driver(). namespace tflite { -class LayerByLayerProfiler : public MicroProfiler { +class LayerByLayerProfiler : public MicroProfilerInterface { public: enum Backend { PRINTF, EVENT_RECORDER }; |