Searched refs:k_y (Results 1 – 7 of 7) sorted by relevance
135 for (int k_y = k_y_start; k_y < k_y_end; k_y++) in arm_avgpool_s16() local139 const int16_t *src_inner = src_base + (ch_src * (k_x + k_y * input_x)); in arm_avgpool_s16()229 for (int k_y = kernel_y_start; k_y < kernel_y_end; k_y++) in arm_avgpool_s16() local233 … const int16_t *start = src + ch_src * (k_x + idx_x + (k_y + idx_y) * input_x); in arm_avgpool_s16()293 for (int k_y = ker_y_start; k_y < kernel_y_end; k_y++) in arm_avgpool_s16() local297 … sum += src[i_ch_in + ch_src * (k_x + base_idx_x + (k_y + base_idx_y) * input_x)]; in arm_avgpool_s16()
138 for (int k_y = k_y_start; k_y < k_y_end; k_y++) in arm_avgpool_s8() local142 const int8_t *src_inner = src_base + (ch_src * (k_x + k_y * input_x)); in arm_avgpool_s8()293 for (int k_y = kernel_y_start; k_y < kernel_y_end; k_y++) in arm_avgpool_s8() local297 … const int8_t *start = src + ch_src * (k_x + idx_x + (k_y + idx_y) * input_x); in arm_avgpool_s8()349 … 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() local353 if (k_y >= 0 && k_x >= 0 && k_y < input_y && k_x < input_x) in arm_avgpool_s8()355 sum += src[i_ch_in + ch_src * (k_x + k_y * input_x)]; in arm_avgpool_s8()
194 for (int k_y = ker_y_start; k_y < kernel_y_end; k_y++) in arm_max_pool_s16() local198 … const int16_t *start = src + channel_in * (k_x + base_idx_x + (k_y + base_idx_y) * input_x); in arm_max_pool_s16()
210 for (int k_y = ker_y_start; k_y < kernel_y_end; k_y++) in arm_max_pool_s8() local214 … const int8_t *start = src + channel_in * (k_x + base_idx_x + (k_y + base_idx_y) * input_x); in arm_max_pool_s8()
117 const int32_t k_y = base_idx_y + dilation_y * i_ker_y; in arm_convolve_s4() local120 if (k_y < 0 || k_y >= input_y || k_x < 0 || k_x >= input_x) in arm_convolve_s4()126 … arm_memcpy_s8(im2col_buf, input_data + (k_y * input_x + k_x) * input_ch, input_ch); in arm_convolve_s4()201 const int32_t k_y = base_idx_y + dilation_y * i_ker_y; in arm_convolve_s4()204 if (k_y < 0 || k_y >= input_y || k_x < 0 || k_x >= input_x) in arm_convolve_s4()213 … input_data + (k_y * input_x + k_x) * input_ch, two_column_buf, input_ch, input_offset); in arm_convolve_s4()
116 const int32_t k_y = base_idx_y + dilation_y * i_ker_y; in arm_convolve_s16() local119 if (k_y < 0 || k_y >= input_y || k_x < 0 || k_x >= input_x) in arm_convolve_s16()127 … (const int8_t *)(input_data + (k_y * input_x + k_x) * input_ch), in arm_convolve_s16()
141 const int32_t k_y = base_idx_y + dilation_y * i_ker_y; in arm_convolve_s8() local144 if (k_y < 0 || k_y >= input_y || k_x < 0 || k_x >= input_x) in arm_convolve_s8()151 … input_data + (k_y * input_x + k_x) * input_ch + i_group * kernel_ch, in arm_convolve_s8()
