aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/NeonInterceptorScheduler.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/neon/NeonInterceptorScheduler.hpp')
-rw-r--r--src/backends/neon/NeonInterceptorScheduler.hpp6
1 files changed, 6 insertions, 0 deletions
diff --git a/src/backends/neon/NeonInterceptorScheduler.hpp b/src/backends/neon/NeonInterceptorScheduler.hpp
index 3a6f2aa5ec..fe80c62c5f 100644
--- a/src/backends/neon/NeonInterceptorScheduler.hpp
+++ b/src/backends/neon/NeonInterceptorScheduler.hpp
@@ -10,6 +10,7 @@
#include <arm_compute/runtime/IScheduler.h>
#include <arm_compute/runtime/Scheduler.h>
#include <arm_compute/core/CPP/ICPPKernel.h>
+#include <arm_compute/core/experimental/Types.h>
namespace armnn
{
@@ -32,6 +33,11 @@ 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,
+ const arm_compute::InputTensorMap &inputs,
+ const arm_compute::OutputTensorMap &outputs ) override;
private:
NeonTimer::KernelMeasurements* m_Kernels;
arm_compute::IScheduler& m_RealScheduler;