diff options
Diffstat (limited to 'src/cpu/kernels/softmax/list.h')
| -rw-r--r-- | src/cpu/kernels/softmax/list.h | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/src/cpu/kernels/softmax/list.h b/src/cpu/kernels/softmax/list.h index ed3515f417..627ce0c264 100644 --- a/src/cpu/kernels/softmax/list.h +++ b/src/cpu/kernels/softmax/list.h @@ -28,9 +28,9 @@ namespace arm_compute { namespace cpu { -#define DECLARE_SOFTMAX_KERNEL(func_name) \ - void func_name(const ITensor *in, const ITensor *max, void *const tmp, \ - ITensor *out, const float beta, bool is_log, const Window &window) +#define DECLARE_SOFTMAX_KERNEL(func_name) \ + void func_name(const ITensor *in, const ITensor *max, void *const tmp, ITensor *out, const float beta, \ + bool is_log, const Window &window) DECLARE_SOFTMAX_KERNEL(neon_fp32_softmax); DECLARE_SOFTMAX_KERNEL(neon_fp16_softmax); @@ -43,8 +43,7 @@ DECLARE_SOFTMAX_KERNEL(sve2_qasymm8_softmax); #undef DECLARE_SOFTMAX_KERNEL -#define DECLARE_LOGITS_KERNEL(func_name) \ - void func_name(const ITensor *in, ITensor *out, const Window &window) +#define DECLARE_LOGITS_KERNEL(func_name) void func_name(const ITensor *in, ITensor *out, const Window &window) DECLARE_LOGITS_KERNEL(neon_fp32_logits); DECLARE_LOGITS_KERNEL(neon_fp16_logits); |