Home
last modified time | relevance | path

Searched refs:base_idx_y (Results 1 – 12 of 12) sorted by relevance

/cmsis-nn-latest/Source/ConvolutionFunctions/
Darm_depthwise_conv_s4.c82 const int16_t base_idx_y = (i_out_y * stride_y) - pad_y; in depthwise_conv_s4_generic() local
117 … const int32_t start_y_max = (-base_idx_y + dilation_y - 1) / dilation_y; in depthwise_conv_s4_generic()
119 … const int32_t end_min_y = (input_y - base_idx_y + dilation_y - 1) / dilation_y; in depthwise_conv_s4_generic()
124 ker_y_start = MAX(0, -base_idx_y); in depthwise_conv_s4_generic()
125 ker_y_end = MIN(kernel_y, input_y - base_idx_y); in depthwise_conv_s4_generic()
134 int32_t idx_y = base_idx_y + dilation_y * ker_y_start; in depthwise_conv_s4_generic()
209 … const int32_t start_y_max = (-base_idx_y + dilation_y - 1) / dilation_y; in depthwise_conv_s4_generic()
211 … const int32_t end_min_y = (input_y - base_idx_y + dilation_y - 1) / dilation_y; in depthwise_conv_s4_generic()
216 ker_y_start = MAX(0, -base_idx_y); in depthwise_conv_s4_generic()
217 ker_y_end = MIN(kernel_y, input_y - base_idx_y); in depthwise_conv_s4_generic()
[all …]
Darm_depthwise_conv_s16.c162 const int16_t base_idx_y = (i_out_y * stride_y) - pad_y; in depthwise_conv_s16_generic_s16() local
195 const int32_t start_y_max = (-base_idx_y + dilation_y - 1) / dilation_y; in depthwise_conv_s16_generic_s16()
197 … const int32_t end_min_y = (input_y - base_idx_y + dilation_y - 1) / dilation_y; in depthwise_conv_s16_generic_s16()
202 ker_y_start = MAX(0, -base_idx_y); in depthwise_conv_s16_generic_s16()
203 ker_y_end = MIN(kernel_y, input_y - base_idx_y); in depthwise_conv_s16_generic_s16()
213 const int32_t idx_y = base_idx_y + dilation_y * i_ker_y; in depthwise_conv_s16_generic_s16()
Darm_depthwise_conv_s8_opt.c113 … for (int i_out_y = 0, base_idx_y = -pad_y; i_out_y < output_y; base_idx_y += stride_y, i_out_y++) in arm_depthwise_conv_s8_opt() local
117 for (int i_ker_y = base_idx_y; i_ker_y < base_idx_y + kernel_y; i_ker_y++) in arm_depthwise_conv_s8_opt()
222 const int16_t base_idx_y = (i_out_y * stride_y) - pad_y; in arm_depthwise_conv_s8_opt() local
229 const int ker_y_start = MAX(0, -base_idx_y); in arm_depthwise_conv_s8_opt()
231 const int ker_y_end = MIN(kernel_y, input_y - base_idx_y); in arm_depthwise_conv_s8_opt()
242 const int32_t idx_y = base_idx_y + i_ker_y; in arm_depthwise_conv_s8_opt()
Darm_depthwise_conv_fast_s16.c110 … for (int i_out_y = 0, base_idx_y = -pad_y; i_out_y < output_y; base_idx_y += stride_y, i_out_y++) in arm_depthwise_conv_fast_s16() local
114 for (int i_ker_y = base_idx_y; i_ker_y < base_idx_y + kernel_y; i_ker_y++) in arm_depthwise_conv_fast_s16()
233 const int16_t base_idx_y = (i_out_y * stride_y) - pad_y; in arm_depthwise_conv_fast_s16() local
240 const int ker_y_start = MAX(0, -base_idx_y); in arm_depthwise_conv_fast_s16()
242 const int ker_y_end = MIN(kernel_y, input_y - base_idx_y); in arm_depthwise_conv_fast_s16()
253 const int32_t idx_y = base_idx_y + i_ker_y; in arm_depthwise_conv_fast_s16()
Darm_depthwise_conv_s8.c190 const int16_t base_idx_y = (i_out_y * stride_y) - pad_y; in depthwise_conv_s8_generic() local
221 const int32_t start_y_max = (-base_idx_y + dilation_y - 1) / dilation_y; in depthwise_conv_s8_generic()
223 … const int32_t end_min_y = (input_y - base_idx_y + dilation_y - 1) / dilation_y; in depthwise_conv_s8_generic()
228 ker_y_start = MAX(0, -base_idx_y); in depthwise_conv_s8_generic()
229 ker_y_end = MIN(kernel_y, input_y - base_idx_y); in depthwise_conv_s8_generic()
239 const int32_t idx_y = base_idx_y + dilation_y * i_ker_y; in depthwise_conv_s8_generic()
Darm_convolve_s4.c111 const int32_t base_idx_y = stride_y * i_out_y - pad_y; in arm_convolve_s4() local
117 const int32_t k_y = base_idx_y + dilation_y * i_ker_y; in arm_convolve_s4()
195 const int32_t base_idx_y = stride_y * i_out_y - pad_y; in arm_convolve_s4()
201 const int32_t k_y = base_idx_y + dilation_y * i_ker_y; in arm_convolve_s4()
Darm_depthwise_conv_s4_opt.c114 … for (int i_out_y = 0, base_idx_y = -pad_y; i_out_y < output_y; base_idx_y += stride_y, i_out_y++) in arm_depthwise_conv_s4_opt() local
118 for (int i_ker_y = base_idx_y; i_ker_y < base_idx_y + kernel_y; i_ker_y++) in arm_depthwise_conv_s4_opt()
264 const int16_t base_idx_y = (i_out_y * stride_y) - pad_y; in arm_depthwise_conv_s4_opt() local
271 const int ker_y_start = MAX(0, -base_idx_y); in arm_depthwise_conv_s4_opt()
273 const int ker_y_end = MIN(kernel_y, input_y - base_idx_y); in arm_depthwise_conv_s4_opt()
284 const int32_t idx_y = base_idx_y + i_ker_y; in arm_depthwise_conv_s4_opt()
Darm_convolve_s16.c110 const int32_t base_idx_y = stride_y * i_out_y - pad_y; in arm_convolve_s16() local
116 const int32_t k_y = base_idx_y + dilation_y * i_ker_y; in arm_convolve_s16()
Darm_convolve_s8.c135 const int32_t base_idx_y = stride_y * i_out_y - pad_y; in arm_convolve_s8() local
141 const int32_t k_y = base_idx_y + dilation_y * i_ker_y; in arm_convolve_s8()
/cmsis-nn-latest/Source/PoolingFunctions/
Darm_max_pool_s16.c180 for (int i_y = 0, base_idx_y = -pad_y; i_y < output_y; base_idx_y += stride_y, i_y++) in arm_max_pool_s16() local
185 const int32_t ker_y_start = MAX(0, -base_idx_y); in arm_max_pool_s16()
189 const int32_t kernel_y_end = MIN(kernel_y, input_y - base_idx_y); in arm_max_pool_s16()
198 … const int16_t *start = src + channel_in * (k_x + base_idx_x + (k_y + base_idx_y) * input_x); in arm_max_pool_s16()
Darm_max_pool_s8.c196 for (int i_y = 0, base_idx_y = -pad_y; i_y < output_y; base_idx_y += stride_y, i_y++) in arm_max_pool_s8() local
201 const int32_t ker_y_start = MAX(0, -base_idx_y); in arm_max_pool_s8()
205 const int32_t kernel_y_end = MIN(kernel_y, input_y - base_idx_y); in arm_max_pool_s8()
214 … const int8_t *start = src + channel_in * (k_x + base_idx_x + (k_y + base_idx_y) * input_x); in arm_max_pool_s8()
Darm_avgpool_s16.c276 for (int i_y = 0, base_idx_y = -pad_y; i_y < output_y; base_idx_y += stride_y, i_y++) in arm_avgpool_s16() local
281 const int32_t ker_y_start = MAX(0, -base_idx_y); in arm_avgpool_s16()
285 const int32_t kernel_y_end = MIN(kernel_y, input_y - base_idx_y); in arm_avgpool_s16()
297 … sum += src[i_ch_in + ch_src * (k_x + base_idx_x + (k_y + base_idx_y) * input_x)]; in arm_avgpool_s16()