aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/NEON/kernels/arm64/NEGEMMLowpAArch64Kernel.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/core/NEON/kernels/arm64/NEGEMMLowpAArch64Kernel.h')
-rw-r--r--arm_compute/core/NEON/kernels/arm64/NEGEMMLowpAArch64Kernel.h8
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 */