Searched refs:idx_out_ch (Results 1 – 3 of 3) sorted by relevance
/cmsis-nn-latest/Source/ConvolutionFunctions/ |
D | arm_depthwise_conv_s4.c | 181 const int idx_out_ch = i_ch_mult + i_input_ch * ch_mult; in depthwise_conv_s4_generic() local 182 if (idx_out_ch && (idx_out_ch % 2 == 0)) in depthwise_conv_s4_generic() 222 acc_0 = bias[idx_out_ch]; in depthwise_conv_s4_generic() 259 … acc_0 = arm_nn_requantize(acc_0, output_mult[idx_out_ch], output_shift[idx_out_ch]); in depthwise_conv_s4_generic() 276 const int idx_out_ch = i_ch_mult + i_input_ch * ch_mult; in depthwise_conv_s4_generic() local 314 acc_0 = bias[idx_out_ch]; in depthwise_conv_s4_generic() 315 acc_1 = bias[idx_out_ch + 1]; in depthwise_conv_s4_generic() 346 … acc_0 = arm_nn_requantize(acc_0, output_mult[idx_out_ch], output_shift[idx_out_ch]); in depthwise_conv_s4_generic() 353 … arm_nn_requantize(acc_1, output_mult[idx_out_ch + 1], output_shift[idx_out_ch + 1]); in depthwise_conv_s4_generic() 384 const int idx_out_ch = i_ch_mult + i_input_ch * ch_mult; in depthwise_conv_s4_generic() local [all …]
|
D | arm_depthwise_conv_s16.c | 170 const int idx_out_ch = i_ch_mult + i_input_ch * ch_mult; in depthwise_conv_s16_generic_s16() local 172 … const int32_t reduced_multiplier = REDUCE_MULTIPLIER(output_mult[idx_out_ch]); in depthwise_conv_s16_generic_s16() 208 acc_0 = bias[idx_out_ch]; in depthwise_conv_s16_generic_s16() 218 … int32_t ker_idx_0 = (i_ker_y * kernel_x + i_ker_x) * (input_ch * ch_mult) + idx_out_ch; in depthwise_conv_s16_generic_s16() 225 … int32_t result = arm_nn_requantize_s64(acc_0, reduced_multiplier, output_shift[idx_out_ch]); in depthwise_conv_s16_generic_s16()
|
D | arm_depthwise_conv_s8.c | 198 const int idx_out_ch = i_ch_mult + i_input_ch * ch_mult; in depthwise_conv_s8_generic() local 234 acc_0 = bias[idx_out_ch]; in depthwise_conv_s8_generic() 244 … 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() 251 … acc_0 = arm_nn_requantize(acc_0, output_mult[idx_out_ch], output_shift[idx_out_ch]); in depthwise_conv_s8_generic()
|