24.02.1
|
Go to the documentation of this file.
24 #ifndef SRC_CPU_KERNELS_ADDMULADD_LIST
25 #define SRC_CPU_KERNELS_ADDMULADD_LIST
35 #define DECLARE_ADD_MUL_ADD_KERNEL(func_name) \
36 void func_name(const ITensor *input1, const ITensor *input2, const ITensor *bn_mul, const ITensor *bn_add, \
37 ITensor *add_output, ITensor *final_output, ConvertPolicy policy, \
38 const ActivationLayerInfo &act_info, const Window &window)
45 #undef DECLARE_ADD_MUL_ADD_KERNEL
#define DECLARE_ADD_MUL_ADD_KERNEL(func_name)
void add_mul_add_fp16_neon(const ITensor *input1, const ITensor *input2, const ITensor *bn_mul, const ITensor *bn_add, ITensor *add_output, ITensor *final_output, ConvertPolicy policy, const ActivationLayerInfo &act_info, const Window &window)
void add_mul_add_fp32_neon(const ITensor *input1, const ITensor *input2, const ITensor *bn_mul, const ITensor *bn_add, ITensor *add_output, ITensor *final_output, ConvertPolicy policy, const ActivationLayerInfo &act_info, const Window &window)
void add_mul_add_s8_neon(const ITensor *input1, const ITensor *input2, const ITensor *bn_mul, const ITensor *bn_add, ITensor *add_output, ITensor *final_output, ConvertPolicy policy, const ActivationLayerInfo &act_info, const Window &window)
void add_mul_add_u8_neon(const ITensor *input1, const ITensor *input2, const ITensor *bn_mul, const ITensor *bn_add, ITensor *add_output, ITensor *final_output, ConvertPolicy policy, const ActivationLayerInfo &act_info, const Window &window)
Copyright (c) 2017-2024 Arm Limited.