aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/NeonInterceptorScheduler.hpp
diff options
context:
space:
mode:
authorNikhil Raj <nikhil.raj@arm.com>2021-01-22 10:02:38 +0000
committerNikhil Raj <nikhil.raj@arm.com>2021-01-25 16:05:12 +0000
commite5617954db782628ca49919a627d01ee0088fb67 (patch)
treeb72cbd4397438887404af486180b3035c01baba1 /src/backends/neon/NeonInterceptorScheduler.hpp
parent5dc64fe99227efece77eb7cd14abc47474a48fb8 (diff)
downloadarmnn-e5617954db782628ca49919a627d01ee0088fb67.tar.gz
Update ACL pin to 6a4ebe1305b544aec1ba0bfc67ed65d94fcc8c2e
* Pass in new window parameter to the schedule_op function Signed-off-by: Nikhil Raj <nikhil.raj@arm.com> Change-Id: I8579d2d6e55ab9888471bd780628df0f73438498
Diffstat (limited to 'src/backends/neon/NeonInterceptorScheduler.hpp')
-rw-r--r--src/backends/neon/NeonInterceptorScheduler.hpp7
1 files changed, 4 insertions, 3 deletions
diff --git a/src/backends/neon/NeonInterceptorScheduler.hpp b/src/backends/neon/NeonInterceptorScheduler.hpp
index f1d1578c8e..aeccca26e2 100644
--- a/src/backends/neon/NeonInterceptorScheduler.hpp
+++ b/src/backends/neon/NeonInterceptorScheduler.hpp
@@ -33,9 +33,10 @@ public:
void SetKernels(NeonTimer::KernelMeasurements* kernels) { m_Kernels = kernels; }
NeonTimer::KernelMeasurements* GetKernels() { return m_Kernels; }
- void schedule_op(arm_compute::ICPPKernel *kernel,
- const Hints &hints,
- arm_compute::ITensorPack &tensors ) override;
+ void schedule_op(arm_compute::ICPPKernel* kernel,
+ const Hints& hints,
+ const arm_compute::Window& window,
+ arm_compute::ITensorPack& tensors ) override;
private:
NeonTimer::KernelMeasurements* m_Kernels;
arm_compute::IScheduler& m_RealScheduler;