diff options
Diffstat (limited to 'src/armnn/NeonInterceptorScheduler.hpp')
-rw-r--r-- | src/armnn/NeonInterceptorScheduler.hpp | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/src/armnn/NeonInterceptorScheduler.hpp b/src/armnn/NeonInterceptorScheduler.hpp index 37966b8178..f33b79a2da 100644 --- a/src/armnn/NeonInterceptorScheduler.hpp +++ b/src/armnn/NeonInterceptorScheduler.hpp @@ -17,7 +17,7 @@ namespace armnn class NeonInterceptorScheduler : public arm_compute::IScheduler { public: - NeonInterceptorScheduler(NeonTimer::KernelMeasurements &kernels, arm_compute::IScheduler &realScheduler); + NeonInterceptorScheduler(arm_compute::IScheduler &realScheduler); ~NeonInterceptorScheduler() = default; void set_num_threads(unsigned int numThreads) override; @@ -28,10 +28,11 @@ public: void run_workloads(std::vector<Workload> &workloads) override; + void SetKernels(NeonTimer::KernelMeasurements* kernels) { m_Kernels = kernels; } + NeonTimer::KernelMeasurements* GetKernels() { return m_Kernels; } private: - NeonTimer::KernelMeasurements& m_Kernels; + NeonTimer::KernelMeasurements* m_Kernels; arm_compute::IScheduler& m_RealScheduler; - WallClockTimer m_Timer; }; } // namespace armnn |