diff options
Diffstat (limited to 'arm_compute/runtime/NEON/INEOperator.h')
-rw-r--r-- | arm_compute/runtime/NEON/INEOperator.h | 14 |
1 files changed, 9 insertions, 5 deletions
diff --git a/arm_compute/runtime/NEON/INEOperator.h b/arm_compute/runtime/NEON/INEOperator.h index a5ffc74940..7971168d24 100644 --- a/arm_compute/runtime/NEON/INEOperator.h +++ b/arm_compute/runtime/NEON/INEOperator.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2020 Arm Limited. + * Copyright (c) 2020-2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -24,20 +24,22 @@ #ifndef ARM_COMPUTE_INEOPERATOR_H #define ARM_COMPUTE_INEOPERATOR_H -#include "../../core/ITensor.h" #include "arm_compute/runtime/IOperator.h" #include "arm_compute/runtime/IRuntimeContext.h" #include "arm_compute/runtime/Types.h" +#include "../../core/ITensor.h" #include <memory> namespace arm_compute { class ICPPKernel; +class Window; + using INEKernel = ICPPKernel; namespace experimental { -/** Basic interface for functions which have a single async NEON kernel */ +/** Basic interface for functions which have a single async CPU kernel */ class INEOperator : public IOperator { public: @@ -58,11 +60,13 @@ public: ~INEOperator(); // Inherited methods overridden: - void run(ITensorPack &tensors) override; - void prepare(ITensorPack &constants) override; + void run(ITensorPack &tensors) override; + void prepare(ITensorPack &constants) override; MemoryRequirements workspace() const override; protected: + void run(ITensorPack &tensors, const Window &window); + std::unique_ptr<INEKernel> _kernel; IRuntimeContext *_ctx; MemoryRequirements _workspace; |