| /cmsis-nn-latest/Source/ConvolutionFunctions/ |
| D | arm_depthwise_conv_s4.c | 85 const int16_t base_idx_x = (i_out_x * stride_x) - pad_x; in depthwise_conv_s4_generic() local
104 … const int32_t start_x_max = (-base_idx_x + dilation_x - 1) / dilation_x; in depthwise_conv_s4_generic()
106 … const int32_t end_min_x = (input_x - base_idx_x + dilation_x - 1) / dilation_x; in depthwise_conv_s4_generic()
111 ker_x_start = MAX(0, -base_idx_x); in depthwise_conv_s4_generic()
112 ker_x_end = MIN(kernel_x, input_x - base_idx_x); in depthwise_conv_s4_generic()
137 int32_t idx_x = base_idx_x + dilation_x * ker_x_start; in depthwise_conv_s4_generic()
196 … const int32_t start_x_max = (-base_idx_x + dilation_x - 1) / dilation_x; in depthwise_conv_s4_generic()
198 … const int32_t end_min_x = (input_x - base_idx_x + dilation_x - 1) / dilation_x; in depthwise_conv_s4_generic()
203 ker_x_start = MAX(0, -base_idx_x); in depthwise_conv_s4_generic()
204 ker_x_end = MIN(kernel_x, input_x - base_idx_x); in depthwise_conv_s4_generic()
[all …]
|
| D | arm_depthwise_conv_s16.c | 165 const int16_t base_idx_x = (i_out_x * stride_x) - pad_x; in depthwise_conv_s16_generic_s16() local
182 const int32_t start_x_max = (-base_idx_x + dilation_x - 1) / dilation_x; in depthwise_conv_s16_generic_s16()
184 … const int32_t end_min_x = (input_x - base_idx_x + dilation_x - 1) / dilation_x; in depthwise_conv_s16_generic_s16()
189 ker_x_start = MAX(0, -base_idx_x); in depthwise_conv_s16_generic_s16()
190 ker_x_end = MIN(kernel_x, input_x - base_idx_x); in depthwise_conv_s16_generic_s16()
216 const int32_t idx_x = base_idx_x + dilation_x * i_ker_x; in depthwise_conv_s16_generic_s16()
|
| D | arm_depthwise_conv_s8.c | 193 const int16_t base_idx_x = (i_out_x * stride_x) - pad_x; in depthwise_conv_s8_generic() local
208 const int32_t start_x_max = (-base_idx_x + dilation_x - 1) / dilation_x; in depthwise_conv_s8_generic()
210 … const int32_t end_min_x = (input_x - base_idx_x + dilation_x - 1) / dilation_x; in depthwise_conv_s8_generic()
215 ker_x_start = MAX(0, -base_idx_x); in depthwise_conv_s8_generic()
216 ker_x_end = MIN(kernel_x, input_x - base_idx_x); in depthwise_conv_s8_generic()
242 const int32_t idx_x = base_idx_x + dilation_x * i_ker_x; in depthwise_conv_s8_generic()
|
| D | arm_convolve_s4.c | 110 const int32_t base_idx_x = stride_x * i_out_x - pad_x; in arm_convolve_s4() local
118 const int32_t k_x = base_idx_x + dilation_x * i_ker_x; in arm_convolve_s4()
194 const int32_t base_idx_x = stride_x * i_out_x - pad_x; in arm_convolve_s4()
202 const int32_t k_x = base_idx_x + dilation_x * i_ker_x; in arm_convolve_s4()
|
| D | arm_depthwise_conv_s8_opt.c | 115 … for (int i_out_x = 0, base_idx_x = -pad_x; i_out_x < output_x; base_idx_x += stride_x, i_out_x++) in arm_depthwise_conv_s8_opt() local
119 for (int i_ker_x = base_idx_x; i_ker_x < base_idx_x + kernel_x; i_ker_x++) in arm_depthwise_conv_s8_opt()
225 const int16_t base_idx_x = (i_out_x * stride_x) - pad_x; in arm_depthwise_conv_s8_opt() local
246 const int32_t idx_x = base_idx_x + i_ker_x; in arm_depthwise_conv_s8_opt()
|
| D | arm_depthwise_conv_fast_s16.c | 112 … for (int i_out_x = 0, base_idx_x = -pad_x; i_out_x < output_x; base_idx_x += stride_x, i_out_x++) in arm_depthwise_conv_fast_s16() local
116 for (int i_ker_x = base_idx_x; i_ker_x < base_idx_x + kernel_x; i_ker_x++) in arm_depthwise_conv_fast_s16()
236 const int16_t base_idx_x = (i_out_x * stride_x) - pad_x; in arm_depthwise_conv_fast_s16() local
257 const int32_t idx_x = base_idx_x + i_ker_x; in arm_depthwise_conv_fast_s16()
|
| D | arm_depthwise_conv_s4_opt.c | 116 … for (int i_out_x = 0, base_idx_x = -pad_x; i_out_x < output_x; base_idx_x += stride_x, i_out_x++) in arm_depthwise_conv_s4_opt() local
120 for (int i_ker_x = base_idx_x; i_ker_x < base_idx_x + kernel_x; i_ker_x++) in arm_depthwise_conv_s4_opt()
267 const int16_t base_idx_x = (i_out_x * stride_x) - pad_x; in arm_depthwise_conv_s4_opt() local
288 const int32_t idx_x = base_idx_x + i_ker_x; in arm_depthwise_conv_s4_opt()
|
| D | arm_convolve_s16.c | 109 const int32_t base_idx_x = stride_x * i_out_x - pad_x; in arm_convolve_s16() local
117 const int32_t k_x = base_idx_x + dilation_x * i_ker_x; in arm_convolve_s16()
|
| D | arm_convolve_s8.c | 134 const int32_t base_idx_x = stride_x * i_out_x - pad_x; in arm_convolve_s8() local
142 const int32_t k_x = base_idx_x + dilation_x * i_ker_x; in arm_convolve_s8()
|
| /cmsis-nn-latest/Source/PoolingFunctions/ |
| D | arm_max_pool_s16.c | 182 for (int i_x = 0, base_idx_x = -pad_x; i_x < output_x; base_idx_x += stride_x, i_x++) in arm_max_pool_s16() local
186 const int32_t ker_x_start = MAX(0, -base_idx_x); in arm_max_pool_s16()
190 const int32_t kernel_x_end = MIN(kernel_x, input_x - base_idx_x); 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()
|
| D | arm_max_pool_s8.c | 198 for (int i_x = 0, base_idx_x = -pad_x; i_x < output_x; base_idx_x += stride_x, i_x++) in arm_max_pool_s8() local
202 const int32_t ker_x_start = MAX(0, -base_idx_x); in arm_max_pool_s8()
206 const int32_t kernel_x_end = MIN(kernel_x, input_x - base_idx_x); 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()
|
| D | arm_avgpool_s16.c | 278 for (int i_x = 0, base_idx_x = -pad_x; i_x < output_x; base_idx_x += stride_x, i_x++) in arm_avgpool_s16() local
282 const int32_t ker_x_start = MAX(0, -base_idx_x); in arm_avgpool_s16()
286 const int32_t kernel_x_end = MIN(kernel_x, input_x - base_idx_x); 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()
|
