diff options
Diffstat (limited to 'arm_compute/core/NEON/kernels/arm64')
-rw-r--r-- | arm_compute/core/NEON/kernels/arm64/NEGEMMLowpAArch64A53Kernel.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/arm_compute/core/NEON/kernels/arm64/NEGEMMLowpAArch64A53Kernel.h b/arm_compute/core/NEON/kernels/arm64/NEGEMMLowpAArch64A53Kernel.h index 6f4ddea854..33cd2d42d0 100644 --- a/arm_compute/core/NEON/kernels/arm64/NEGEMMLowpAArch64A53Kernel.h +++ b/arm_compute/core/NEON/kernels/arm64/NEGEMMLowpAArch64A53Kernel.h @@ -37,11 +37,18 @@ class ITensor; class NEGEMMLowpAArch64A53Kernel : public NEGEMMAssemblyBaseKernel { public: + /** Default constructor */ + NEGEMMLowpAArch64A53Kernel(); // Inherited methods overridden: void run(const Window &window, const ThreadInfo &info) override; protected: void internal_configure(const ITensor *input0, const ITensor *input1, ITensor *output, ITensor *workspace, float alpha, float beta, bool transform_0, bool transform_1) override; + +private: + using NEGEMMLowpAArch64A53 = void(const ITensor *input0, const ITensor *input1, ITensor *output, ITensor *workspace, float alpha, float beta, bool transform_0, bool transform_1, const Window &window, + const ThreadInfo &info); + NEGEMMLowpAArch64A53 *_func; }; } // namespace arm_compute #endif /* ARM_COMPUTE_AARCH64_V8A */ |