24.04
|
Go to the documentation of this file.
24 #ifndef ACL_SRC_CPU_KERNELS_SOFTMAX_LIST_H
25 #define ACL_SRC_CPU_KERNELS_SOFTMAX_LIST_H
31 #define DECLARE_SOFTMAX_KERNEL(func_name) \
32 template <bool IS_LOG> \
33 void func_name(const ITensor *in, void *const tmp, ITensor *out, const float beta, int axis, const Window &window)
40 #ifdef ARM_COMPUTE_ENABLE_SME2
42 void sme2_fp32_softmax(
43 const ITensor *in,
void *
const tmp, ITensor *out,
const float beta,
int axis,
const Window &window);
45 void sme2_fp16_softmax(
46 const ITensor *in,
void *
const tmp, ITensor *out,
const float beta,
int axis,
const Window &window);
48 #endif // ARM_COMPUTE_ENABLE_SME2
50 #undef DECLARE_SOFTMAX_KERNEL
54 #endif // ACL_SRC_CPU_KERNELS_SOFTMAX_LIST_H
void neon_fp16_softmax(const ITensor *in, void *const tmp, ITensor *out, const float beta, int axis, const Window &window)
void neon_qasymm8_softmax(const ITensor *in, void *const tmp, ITensor *out, const float beta, int axis, const Window &window)
void neon_fp32_softmax(const ITensor *in, void *const tmp, ITensor *out, const float beta, int axis, const Window &window)
void neon_qasymm8_signed_softmax(const ITensor *in, void *const tmp, ITensor *out, const float beta, int axis, const Window &window)
Copyright (c) 2017-2024 Arm Limited.
#define DECLARE_SOFTMAX_KERNEL(func_name)