diff options
Diffstat (limited to 'arm_compute/core/NEON/kernels/arm64/NEGEMMLowpAArch64Kernel.h')
-rw-r--r-- | arm_compute/core/NEON/kernels/arm64/NEGEMMLowpAArch64Kernel.h | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/arm_compute/core/NEON/kernels/arm64/NEGEMMLowpAArch64Kernel.h b/arm_compute/core/NEON/kernels/arm64/NEGEMMLowpAArch64Kernel.h index e8e71cf3d7..a93df033de 100644 --- a/arm_compute/core/NEON/kernels/arm64/NEGEMMLowpAArch64Kernel.h +++ b/arm_compute/core/NEON/kernels/arm64/NEGEMMLowpAArch64Kernel.h @@ -37,11 +37,19 @@ class ITensor; class NEGEMMLowpAArch64Kernel : public NEGEMMAssemblyBaseKernel { public: + /** Default constructor */ + NEGEMMLowpAArch64Kernel(); + // 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 NEGEMMLowpAArch64 = 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); + NEGEMMLowpAArch64 *_func; }; } // namespace arm_compute #endif /* ARM_COMPUTE_AARCH64_V8A */ |