diff options
Diffstat (limited to 'src/cpu/kernels/softmax/list.h')
-rw-r--r-- | src/cpu/kernels/softmax/list.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/cpu/kernels/softmax/list.h b/src/cpu/kernels/softmax/list.h index 16fbd31a19..1bb8ed50f0 100644 --- a/src/cpu/kernels/softmax/list.h +++ b/src/cpu/kernels/softmax/list.h @@ -42,6 +42,9 @@ DECLARE_SOFTMAX_KERNEL(neon_qasymm8_signed_softmax); void sme2_fp32_softmax( const ITensor *in, void *const tmp, ITensor *out, const float beta, int axis, const Window &window); +void sme2_fp16_softmax( + const ITensor *in, void *const tmp, ITensor *out, const float beta, int axis, const Window &window); + #endif // ARM_COMPUTE_ENABLE_SME2 #undef DECLARE_SOFTMAX_KERNEL |