Searched refs:k_y (Results 1 – 5 of 5) sorted by relevance
122 for (int k_y = k_y_start; k_y < k_y_end; k_y++) in arm_avgpool_s16() local126 const int16_t *src_inner = src_base + (ch_src * (k_x + k_y * input_x)); in arm_avgpool_s16()208 for (int k_y = kernel_y_start; k_y < kernel_y_end; k_y++) in arm_avgpool_s16() local212 const int16_t *start = src + ch_src * (k_x + idx_x + (k_y + idx_y) * input_x); in arm_avgpool_s16()266 for (int k_y = ker_y_start; k_y < kernel_y_end; k_y++) in arm_avgpool_s16() local270 … sum += src[i_ch_in + ch_src * (k_x + base_idx_x + (k_y + base_idx_y) * input_x)]; in arm_avgpool_s16()
128 for (int k_y = k_y_start; k_y < k_y_end; k_y++) in arm_avgpool_s8() local132 const int8_t *src_inner = src_base + (ch_src * (k_x + k_y * input_x)); in arm_avgpool_s8()268 for (int k_y = kernel_y_start; k_y < kernel_y_end; k_y++) in arm_avgpool_s8() local272 const int8_t *start = src + ch_src * (k_x + idx_x + (k_y + idx_y) * input_x); in arm_avgpool_s8()316 … for (int k_y = i_y * stride_y - pad_y; k_y < i_y * stride_y - pad_y + kernel_y; k_y++) in arm_avgpool_s8() local320 if (k_y >= 0 && k_x >= 0 && k_y < input_y && k_x < input_x) in arm_avgpool_s8()322 sum += src[i_ch_in + ch_src * (k_x + k_y * input_x)]; in arm_avgpool_s8()
183 for (int k_y = ker_y_start; k_y < kernel_y_end; k_y++) in arm_max_pool_s16() local187 … const int16_t *start = src + channel_in * (k_x + base_idx_x + (k_y + base_idx_y) * input_x); in arm_max_pool_s16()
201 for (int k_y = ker_y_start; k_y < kernel_y_end; k_y++) in arm_max_pool_s8() local205 … const int8_t *start = src + channel_in * (k_x + base_idx_x + (k_y + base_idx_y) * input_x); in arm_max_pool_s8()
123 const int32_t k_y = base_idx_y + dilation_y * i_ker_y; in arm_convolve_s8() local126 if (k_y < 0 || k_y >= input_y || k_x < 0 || k_x >= input_x) in arm_convolve_s8()132 … arm_memcpy_s8(im2col_buf, input_data + (k_y * input_x + k_x) * input_ch, input_ch); in arm_convolve_s8()
