From 668ccdcfb81bfab3a2d44cd1ddd956e83a2dfb09 Mon Sep 17 00:00:00 2001 From: Sang-Hoon Park Date: Wed, 3 Feb 2021 10:32:59 +0000 Subject: Add dynamic tensor support to CpuElementwise The kernels and operators for binary and unary operations are now capable of being configured with dynamic shapes and computing windows at run-time. Additionally, changing arguments' names is done for consistency. Partially Implements: COMPMID-4127 Change-Id: I48e5038692db667dec7cb2b2906fe5683214fe19 Signed-off-by: Sang-Hoon Park Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4973 Tested-by: Arm Jenkins Reviewed-by: Pablo Marquez Tello Comments-Addressed: Arm Jenkins --- arm_compute/runtime/NEON/INEOperator.h | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'arm_compute/runtime/NEON/INEOperator.h') 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 _kernel; IRuntimeContext *_ctx; MemoryRequirements _workspace; -- cgit v1.2.1