diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/backends/neon/NeonInterceptorScheduler.cpp | 5 | ||||
-rw-r--r-- | src/backends/neon/NeonInterceptorScheduler.hpp | 3 |
2 files changed, 3 insertions, 5 deletions
diff --git a/src/backends/neon/NeonInterceptorScheduler.cpp b/src/backends/neon/NeonInterceptorScheduler.cpp index b4d9b300df..c20be2b359 100644 --- a/src/backends/neon/NeonInterceptorScheduler.cpp +++ b/src/backends/neon/NeonInterceptorScheduler.cpp @@ -54,12 +54,11 @@ void NeonInterceptorScheduler::run_tagged_workloads(std::vector<Workload> &workl void NeonInterceptorScheduler::schedule_op(arm_compute::ICPPKernel *kernel, const Hints &hints, - const arm_compute::InputTensorMap &inputs, - const arm_compute::OutputTensorMap &outputs ) + arm_compute::ITensorPack &tensors ) { WallClockTimer::clock::time_point startTime = WallClockTimer::clock::now(); - m_RealScheduler.schedule_op(kernel, hints, inputs, outputs); + m_RealScheduler.schedule_op(kernel, hints, tensors); WallClockTimer::clock::time_point stopTime = WallClockTimer::clock::now(); const auto delta = std::chrono::duration<double, std::micro>(stopTime - startTime); diff --git a/src/backends/neon/NeonInterceptorScheduler.hpp b/src/backends/neon/NeonInterceptorScheduler.hpp index e01f89bf5c..f1d1578c8e 100644 --- a/src/backends/neon/NeonInterceptorScheduler.hpp +++ b/src/backends/neon/NeonInterceptorScheduler.hpp @@ -35,8 +35,7 @@ public: void schedule_op(arm_compute::ICPPKernel *kernel, const Hints &hints, - const arm_compute::InputTensorMap &inputs, - const arm_compute::OutputTensorMap &outputs ) override; + arm_compute::ITensorPack &tensors ) override; private: NeonTimer::KernelMeasurements* m_Kernels; arm_compute::IScheduler& m_RealScheduler; |