Searched refs:ch_mult (Results 1 – 5 of 5) sorted by relevance
| /hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Source/ConvolutionFunctions/ |
| D | arm_depthwise_conv_u8_basic_ver1.c | 49 const int32_t ch_mult, in depthwise_conv_u8_mult_4() argument
73 ++in_ch, out_ch += ch_mult) in depthwise_conv_u8_mult_4()
75 for (int mult_tile = 0; mult_tile < ch_mult; mult_tile += 4) in depthwise_conv_u8_mult_4()
138 const int32_t ch_mult, in depthwise_conv_u8_generic() argument
167 for (int i_ch_mult = 0; i_ch_mult < ch_mult; i_ch_mult++) in depthwise_conv_u8_generic()
169 const int idx_out_ch = i_ch_mult + i_input_ch * ch_mult; in depthwise_conv_u8_generic()
186 … int32_t ker_idx_0 = (i_ker_y * kernel_x + i_ker_x) * (input_ch * ch_mult) + idx_out_ch; in depthwise_conv_u8_generic()
253 const int16_t ch_mult, in arm_depthwise_conv_u8_basic_ver1() argument
275 if (ch_mult % 4 == 0) in arm_depthwise_conv_u8_basic_ver1()
282 ch_mult * input_ch, in arm_depthwise_conv_u8_basic_ver1()
[all …]
|
| D | arm_depthwise_conv_s8.c | 49 const int32_t ch_mult, in depthwise_conv_s8_mult_4() argument
72 ++in_ch, out_ch += ch_mult) in depthwise_conv_s8_mult_4()
74 for (int mult_tile = 0; mult_tile < ch_mult; mult_tile += 4) in depthwise_conv_s8_mult_4()
149 const uint16_t ch_mult, in depthwise_conv_s8_generic() argument
181 for (int i_ch_mult = 0; i_ch_mult < ch_mult; i_ch_mult++) in depthwise_conv_s8_generic()
183 const int idx_out_ch = i_ch_mult + i_input_ch * ch_mult; in depthwise_conv_s8_generic()
200 … int32_t ker_idx_0 = (i_ker_y * kernel_x + i_ker_x) * (input_ch * ch_mult) + idx_out_ch; in depthwise_conv_s8_generic()
245 if (dw_conv_params->ch_mult % 4 == 0 && input_dims->n == 1) in arm_depthwise_conv_s8()
253 dw_conv_params->ch_mult, in arm_depthwise_conv_s8()
280 dw_conv_params->ch_mult, in arm_depthwise_conv_s8()
|
| D | arm_depthwise_conv_wrapper_s8.c | 62 if (1 == dw_conv_params->ch_mult && input_dims->n == 1) in arm_depthwise_conv_wrapper_s8()
|
| /hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Include/ |
| D | arm_nn_types.h | 97 int32_t ch_mult; /**< Channel Multiplier. ch_mult * in_ch = out_ch */ member
|
| D | arm_nnfunctions.h | 1826 const int16_t ch_mult,
|
