Searched refs:k_x (Results 1 – 7 of 7) sorted by relevance
137 for (int k_x = k_x_start; k_x < k_x_end; k_x++) in arm_avgpool_s16() local139 const int16_t *src_inner = src_base + (ch_src * (k_x + k_y * input_x)); in arm_avgpool_s16()231 for (int k_x = kernel_x_start; k_x < kernel_x_end; k_x++) 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()295 for (int k_x = ker_x_start; k_x < kernel_x_end; k_x++) 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()
140 for (int k_x = k_x_start; k_x < k_x_end; k_x++) in arm_avgpool_s8() local142 const int8_t *src_inner = src_base + (ch_src * (k_x + k_y * input_x)); in arm_avgpool_s8()295 for (int k_x = kernel_x_start; k_x < kernel_x_end; k_x++) 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()351 … for (int k_x = i_x * stride_x - pad_x; k_x < i_x * stride_x - pad_x + kernel_x; k_x++) 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()
196 for (int k_x = ker_x_start; k_x < kernel_x_end; k_x++) 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()
212 for (int k_x = ker_x_start; k_x < kernel_x_end; k_x++) 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()
118 const int32_t k_x = base_idx_x + dilation_x * i_ker_x; 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()202 const int32_t k_x = base_idx_x + dilation_x * i_ker_x; 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()
117 const int32_t k_x = base_idx_x + dilation_x * i_ker_x; 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()
142 const int32_t k_x = base_idx_x + dilation_x * i_ker_x; 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()
