Go to the source code of this file.
|
template<bool IS_LOG> |
void | neon_fp32_softmax (const ITensor *in, void *const tmp, ITensor *out, const float beta, const Window &window) |
|
template<bool IS_LOG> |
void | neon_fp16_softmax (const ITensor *in, void *const tmp, ITensor *out, const float beta, const Window &window) |
|
template<bool IS_LOG> |
void | neon_qasymm8_softmax (const ITensor *in, void *const tmp, ITensor *out, const float beta, const Window &window) |
|
template<bool IS_LOG> |
void | neon_qasymm8_signed_softmax (const ITensor *in, void *const tmp, ITensor *out, const float beta, const Window &window) |
|
◆ DECLARE_SOFTMAX_KERNEL
#define DECLARE_SOFTMAX_KERNEL |
( |
|
func_name | ) |
|
Value: template <bool IS_LOG> \
void func_name(const ITensor *in, void *const tmp, ITensor *out, const float beta, const Window &window)
Definition at line 31 of file list.h.