24.04
|
Go to the documentation of this file.
24 #ifndef SRC_CORE_NEON_KERNELS_L2NORMLAYER_LIST_H
25 #define SRC_CORE_NEON_KERNELS_L2NORMLAYER_LIST_H
30 #define DECLARE_L2NORMLAYER_KERNEL(func_name) \
31 void func_name(const ITensor *in, const ITensor *sum, ITensor *out, float epsilon, const Window &window, \
39 #undef DECLARE_L2NORMLAYER_KERNEL
42 #endif //SRC_CORE_NEON_KERNELS_L2NORMLAYER_LIST_H
#define DECLARE_L2NORMLAYER_KERNEL(func_name)
void neon_fp16_l2_normalize_x(const ITensor *in, const ITensor *sum, ITensor *out, float epsilon, const Window &window, size_t axis)
void neon_fp32_l2_normalize_x(const ITensor *in, const ITensor *sum, ITensor *out, float epsilon, const Window &window, size_t unused_axis)
void neon_fp16_l2_normalize_yz(const ITensor *in, const ITensor *sum, ITensor *out, float epsilon, const Window &window, size_t axis)
Copyright (c) 2017-2024 Arm Limited.
void neon_fp32_l2_normalize_yz(const ITensor *in, const ITensor *sum, ITensor *out, float epsilon, const Window &window, size_t axis)