24.02.1
|
Go to the documentation of this file.
24 #ifndef ACL_SRC_CORE_NEON_KERNELS_BATCHNORMALIZATION_IMPL_LIST_H
25 #define ACL_SRC_CORE_NEON_KERNELS_BATCHNORMALIZATION_IMPL_LIST_H
31 #define DECLARE_BATCH_NORMALIZATION_KERNEL(func_name) \
32 void func_name(ITensor *src, ITensor *dst, const ITensor *mean, const ITensor *var, const ITensor *beta, \
33 const ITensor *gamma, float epsilon, ActivationLayerInfo &act_info, const Window &window)
40 #define DECLARE_BATCH_NORMALIZATION_NCHW_KERNEL(func_name) \
41 void func_name(const Window &window, ITensor *input, ITensor *output, const ITensor *mean, const ITensor *var, \
42 const ITensor *beta, const ITensor *gamma, float epsilon, ActivationLayerInfo act_info)
53 #undef DECLARE_BATCH_NORMALIZATION_KERNEL
54 #undef DECLARE_BATCH_NORMALIZATION_NCHW_KERNEL
59 #endif // ACL_SRC_CORE_NEON_KERNELS_BATCHNORMALIZATION_IMPL_LIST_H
void fp32_batch_normalization_nchw_non_fused_lubrelu(const Window &window, ITensor *input, ITensor *output, const ITensor *mean, const ITensor *var, const ITensor *beta, const ITensor *gamma, float epsilon, ActivationLayerInfo act_info)
void fp32_neon_batch_normalization(ITensor *src, ITensor *dst, const ITensor *mean, const ITensor *var, const ITensor *beta, const ITensor *gamma, float epsilon, ActivationLayerInfo &act_info, const Window &window)
void fp32_batch_normalization_nchw_non_fused(const Window &window, ITensor *input, ITensor *output, const ITensor *mean, const ITensor *var, const ITensor *beta, const ITensor *gamma, float epsilon, ActivationLayerInfo act_info)
void fp16_sve_batch_normalization(ITensor *src, ITensor *dst, const ITensor *mean, const ITensor *var, const ITensor *beta, const ITensor *gamma, float epsilon, ActivationLayerInfo &act_info, const Window &window)
void fp16_batch_normalization_nchw_non_fused(const Window &window, ITensor *input, ITensor *output, const ITensor *mean, const ITensor *var, const ITensor *beta, const ITensor *gamma, float epsilon, ActivationLayerInfo act_info)
void fp16_neon_batch_normalization(ITensor *src, ITensor *dst, const ITensor *mean, const ITensor *var, const ITensor *beta, const ITensor *gamma, float epsilon, ActivationLayerInfo &act_info, const Window &window)
void fp16_batch_normalization_nchw_non_fused_brelu(const Window &window, ITensor *input, ITensor *output, const ITensor *mean, const ITensor *var, const ITensor *beta, const ITensor *gamma, float epsilon, ActivationLayerInfo act_info)
void fp16_batch_normalization_nchw_non_fused_relu(const Window &window, ITensor *input, ITensor *output, const ITensor *mean, const ITensor *var, const ITensor *beta, const ITensor *gamma, float epsilon, ActivationLayerInfo act_info)
void fp32_batch_normalization_nchw_non_fused_relu(const Window &window, ITensor *input, ITensor *output, const ITensor *mean, const ITensor *var, const ITensor *beta, const ITensor *gamma, float epsilon, ActivationLayerInfo act_info)
Copyright (c) 2017-2024 Arm Limited.
void fp32_sve_batch_normalization(ITensor *src, ITensor *dst, const ITensor *mean, const ITensor *var, const ITensor *beta, const ITensor *gamma, float epsilon, ActivationLayerInfo &act_info, const Window &window)
#define DECLARE_BATCH_NORMALIZATION_NCHW_KERNEL(func_name)
void fp16_batch_normalization_nchw_non_fused_lubrelu(const Window &window, ITensor *input, ITensor *output, const ITensor *mean, const ITensor *var, const ITensor *beta, const ITensor *gamma, float epsilon, ActivationLayerInfo act_info)
#define DECLARE_BATCH_NORMALIZATION_KERNEL(func_name)
void fp32_batch_normalization_nchw_non_fused_brelu(const Window &window, ITensor *input, ITensor *output, const ITensor *mean, const ITensor *var, const ITensor *beta, const ITensor *gamma, float epsilon, ActivationLayerInfo act_info)