Lines Matching refs:ch_src
95 const int32_t ch_src = input_dims->c; in arm_avgpool_s8() local
96 const int32_t batch_input = input_x * input_y * ch_src; in arm_avgpool_s8()
97 const int32_t batch_output = output_x * output_y * ch_src; in arm_avgpool_s8()
118 int8_t *out = &dst[ch_src * (i_x + i_y * output_x)]; in arm_avgpool_s8()
120 int32_t ch_count = (ch_src + 15) / 16; in arm_avgpool_s8()
121 int32_t channels = ch_src; in arm_avgpool_s8()
142 const int8_t *src_inner = src_base + (ch_src * (k_x + k_y * input_x)); in arm_avgpool_s8()
256 const int32_t ch_src = input_dims->c; in arm_avgpool_s8() local
272 const int32_t batch_size = input_x * input_y * ch_src; in arm_avgpool_s8()
297 … const int8_t *start = src + ch_src * (k_x + idx_x + (k_y + idx_y) * input_x); in arm_avgpool_s8()
301 for (int i = 0; i < ch_src; i++) in arm_avgpool_s8()
308 for (int i = 0; i < ch_src; i++) in arm_avgpool_s8()
323 scale_q31_to_q7_and_clamp(buffer, dst, ch_src, count, act_min, act_max); in arm_avgpool_s8()
324 dst += ch_src; in arm_avgpool_s8()
336 const int32_t batch_input = input_x * input_y * ch_src; in arm_avgpool_s8()
337 const int32_t batch_output = output_x * output_y * ch_src; in arm_avgpool_s8()
345 for (int i_ch_in = 0; i_ch_in < ch_src; i_ch_in++) in arm_avgpool_s8()
355 sum += src[i_ch_in + ch_src * (k_x + k_y * input_x)]; in arm_avgpool_s8()
371 dst[i_ch_in + ch_src * (i_x + i_y * output_x)] = sum; in arm_avgpool_s8()