Lines Matching refs:input_ch
63 const int32_t input_ch = input_dims->c; in arm_depthwise_conv_s8_opt() local
67 if (input_ch != output_ch) in arm_depthwise_conv_s8_opt()
103 const int32_t ch_loop = (input_ch + (CH_IN_BLOCK_MVE - 1)) / CH_IN_BLOCK_MVE; in arm_depthwise_conv_s8_opt()
128 input_slice + (i_ker_y * input_x + i_ker_x) * input_ch, in arm_depthwise_conv_s8_opt()
145 input_ch, in arm_depthwise_conv_s8_opt()
155 out += (4 * input_ch); in arm_depthwise_conv_s8_opt()
168 out = out_base + (i_buf * input_ch); in arm_depthwise_conv_s8_opt()
188 row_0 += input_ch; in arm_depthwise_conv_s8_opt()
236 … memset(&col_buffer[index], 0, (kernel_x * input_ch) * ker_y_start * sizeof(int16_t)); in arm_depthwise_conv_s8_opt()
237 index += (kernel_x * input_ch) * ker_y_start; in arm_depthwise_conv_s8_opt()
249 memset(&col_buffer[index], 0, input_ch * sizeof(int16_t)); in arm_depthwise_conv_s8_opt()
253 … arm_q7_to_q15_with_offset((int8_t *)input + (idx_y * input_x + idx_x) * input_ch, in arm_depthwise_conv_s8_opt()
255 input_ch, in arm_depthwise_conv_s8_opt()
258 index += input_ch; in arm_depthwise_conv_s8_opt()
265 memset(&col_buffer[index], 0, (kernel_x * input_ch) * diff * sizeof(int16_t)); in arm_depthwise_conv_s8_opt()
301 ip_a1 = arm_nn_read_s8x4(row_pos + input_ch); in arm_depthwise_conv_s8_opt()
303 op_b = arm_nn_read_s16x2(col_pos + input_ch); in arm_depthwise_conv_s8_opt()
320 op_b = arm_nn_read_s16x2(col_pos + input_ch + 2); in arm_depthwise_conv_s8_opt()
330 row_pos += input_ch << 1; in arm_depthwise_conv_s8_opt()
331 col_pos += input_ch << 1; in arm_depthwise_conv_s8_opt()
343 row_pos += input_ch; in arm_depthwise_conv_s8_opt()
344 col_pos += input_ch; in arm_depthwise_conv_s8_opt()
389 sum += row_pos[i * input_ch] * col_pos[i * input_ch]; in arm_depthwise_conv_s8_opt()