diff options
Diffstat (limited to 'src/backends/neon/NeonInterceptorScheduler.hpp')
-rw-r--r-- | src/backends/neon/NeonInterceptorScheduler.hpp | 6 |
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; |