aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/runtime/NEON/INEOperator.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/runtime/NEON/INEOperator.h')
-rw-r--r--arm_compute/runtime/NEON/INEOperator.h4
1 files changed, 4 insertions, 0 deletions
diff --git a/arm_compute/runtime/NEON/INEOperator.h b/arm_compute/runtime/NEON/INEOperator.h
index b21dc49b20..184a5959b4 100644
--- a/arm_compute/runtime/NEON/INEOperator.h
+++ b/arm_compute/runtime/NEON/INEOperator.h
@@ -34,6 +34,8 @@
namespace arm_compute
{
class ICPPKernel;
+class Window;
+
using INEKernel = ICPPKernel;
namespace experimental
{
@@ -63,6 +65,8 @@ public:
MemoryRequirements workspace() const override;
protected:
+ void run(ITensorPack &tensors, const Window &window);
+
std::unique_ptr<INEKernel> _kernel;
IRuntimeContext *_ctx;
MemoryRequirements _workspace;