Searched refs:num_ch (Results 1 – 3 of 3) sorted by relevance
42 const uint16_t num_ch, in arm_nn_depthwise_conv_s8_core() argument53 int32_t ch_per_loop = num_ch / 4; in arm_nn_depthwise_conv_s8_core()72 const int16_t *col_1 = col + kernel_size * num_ch + offset; in arm_nn_depthwise_conv_s8_core()78 const int8_t *row_1 = row_0 + num_ch; in arm_nn_depthwise_conv_s8_core()79 const int8_t *row_2 = row_0 + 2 * num_ch; in arm_nn_depthwise_conv_s8_core()85 col_0 += num_ch; in arm_nn_depthwise_conv_s8_core()86 col_1 += num_ch; in arm_nn_depthwise_conv_s8_core()93 col_0 += num_ch; in arm_nn_depthwise_conv_s8_core()94 col_1 += num_ch; in arm_nn_depthwise_conv_s8_core()101 col_0 += num_ch; in arm_nn_depthwise_conv_s8_core()[all …]
49 const uint16_t num_ch, in arm_nn_depthwise_conv_nt_t_s16() argument61 int32_t loop_count = (num_ch + 3) / 4; in arm_nn_depthwise_conv_nt_t_s16()62 uint32_t num_ch_to_process = num_ch; in arm_nn_depthwise_conv_nt_t_s16()69 const int16_t *lhs_1 = lhs + row_x_col * num_ch + offset; in arm_nn_depthwise_conv_nt_t_s16()70 const int16_t *lhs_2 = lhs + (row_x_col * num_ch * 2) + offset; in arm_nn_depthwise_conv_nt_t_s16()71 const int16_t *lhs_3 = lhs + (row_x_col * num_ch * 3) + offset; in arm_nn_depthwise_conv_nt_t_s16()94 lhs_0 += num_ch; in arm_nn_depthwise_conv_nt_t_s16()95 lhs_1 += num_ch; in arm_nn_depthwise_conv_nt_t_s16()96 lhs_2 += num_ch; in arm_nn_depthwise_conv_nt_t_s16()97 lhs_3 += num_ch; in arm_nn_depthwise_conv_nt_t_s16()[all …]
235 const uint16_t num_ch,875 const uint16_t num_ch,
