| /cmsis-nn-latest/Source/ConvolutionFunctions/ |
| D | arm_depthwise_conv_s4.c | 52 const int32_t kernel_y, in depthwise_conv_s4_generic() argument
120 ker_y_end = MIN(kernel_y, end_min_y); in depthwise_conv_s4_generic()
125 ker_y_end = MIN(kernel_y, input_y - base_idx_y); in depthwise_conv_s4_generic()
212 ker_y_end = MIN(kernel_y, end_min_y); in depthwise_conv_s4_generic()
217 ker_y_end = MIN(kernel_y, input_y - base_idx_y); in depthwise_conv_s4_generic()
304 ker_y_end = MIN(kernel_y, end_min_y); in depthwise_conv_s4_generic()
309 ker_y_end = MIN(kernel_y, input_y - base_idx_y); in depthwise_conv_s4_generic()
416 ker_y_end = MIN(kernel_y, end_min_y); in depthwise_conv_s4_generic()
421 ker_y_end = MIN(kernel_y, input_y - base_idx_y); in depthwise_conv_s4_generic()
461 if ((kernel_x * kernel_y) % 2) in depthwise_conv_s4_generic()
|
| D | arm_depthwise_conv_s8_opt.c | 81 const int32_t kernel_y = filter_dims->h; in arm_depthwise_conv_s8_opt() local
101 const int32_t kernel_size = kernel_x * kernel_y; in arm_depthwise_conv_s8_opt()
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()
231 const int ker_y_end = MIN(kernel_y, input_y - base_idx_y); in arm_depthwise_conv_s8_opt()
262 const int diff = kernel_y - ker_y_end; in arm_depthwise_conv_s8_opt()
288 uint16_t col_count = (kernel_x * kernel_y) / 2; in arm_depthwise_conv_s8_opt()
335 col_count = (kernel_x * kernel_y) & 0x1; in arm_depthwise_conv_s8_opt()
384 const uint16_t col_count = (kernel_x * kernel_y); in arm_depthwise_conv_s8_opt()
|
| D | arm_depthwise_conv_fast_s16.c | 88 const int32_t kernel_y = filter_dims->h; in arm_depthwise_conv_fast_s16() local
105 const int32_t kernel_size = kernel_x * kernel_y; in arm_depthwise_conv_fast_s16()
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()
242 const int ker_y_end = MIN(kernel_y, input_y - base_idx_y); in arm_depthwise_conv_fast_s16()
273 const int diff = kernel_y - ker_y_end; in arm_depthwise_conv_fast_s16()
298 uint16_t col_count = (kernel_x * kernel_y) / 2; in arm_depthwise_conv_fast_s16()
345 col_count = (kernel_x * kernel_y) & 0x1; in arm_depthwise_conv_fast_s16()
401 const uint16_t col_count = (kernel_x * kernel_y); in arm_depthwise_conv_fast_s16()
|
| D | arm_depthwise_conv_s16.c | 51 const int32_t kernel_y, in depthwise_conv_s16_mult_4_s16() argument
89 … for (int32_t ker_h = ker_h_start; ker_h < MIN(kernel_y, input_y - in_h); ++ker_h) in depthwise_conv_s16_mult_4_s16()
141 const uint16_t kernel_y, in depthwise_conv_s16_generic_s16() argument
198 ker_y_end = MIN(kernel_y, end_min_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()
|
| D | arm_convolve_s4.c | 75 const uint16_t kernel_y = filter_dims->h; in arm_convolve_s4() local
89 const int32_t rhs_cols = kernel_x * kernel_y * input_ch; in arm_convolve_s4()
113 for (int32_t i_ker_y = 0; i_ker_y < kernel_y; i_ker_y++) in arm_convolve_s4()
197 for (int32_t i_ker_y = 0; i_ker_y < kernel_y; i_ker_y++) in arm_convolve_s4()
|
| D | arm_depthwise_conv_s8.c | 55 const int32_t kernel_y, in depthwise_conv_s8_mult_4() argument
97 … for (int32_t ker_h = ker_h_start; ker_h < MIN(kernel_y, input_y - in_h); ++ker_h) in depthwise_conv_s8_mult_4()
163 const uint16_t kernel_y, in depthwise_conv_s8_generic() argument
224 ker_y_end = MIN(kernel_y, end_min_y); in depthwise_conv_s8_generic()
229 ker_y_end = MIN(kernel_y, input_y - base_idx_y); in depthwise_conv_s8_generic()
|
| D | arm_convolve_s16.c | 75 const int32_t kernel_y = filter_dims->h; in arm_convolve_s16() local
79 const int32_t rhs_cols = input_ch * kernel_y * kernel_x; in arm_convolve_s16()
112 for (int32_t i_ker_y = 0; i_ker_y < kernel_y; i_ker_y++) in arm_convolve_s16()
|
| D | arm_transpose_conv_s8.c | 76 const uint16_t kernel_y = filter_dims->h; in arm_transpose_conv_s8() local
147 const int32_t col_y = (output_y + pad_y_offset + pad_y - kernel_y) / stride_y + 1; in arm_transpose_conv_s8()
157 for (int32_t i_ker_y = i_pad_y; i_ker_y < i_pad_y + kernel_y; i_ker_y++) in arm_transpose_conv_s8()
|
| D | arm_convolve_s8.c | 74 const uint16_t kernel_y = filter_dims->h; in arm_convolve_s8() local
92 const int32_t rhs_cols = kernel_x * kernel_y * kernel_ch; in arm_convolve_s8()
137 for (int32_t i_ker_y = 0; i_ker_y < kernel_y; i_ker_y++) in arm_convolve_s8()
|
| D | arm_depthwise_conv_s4_opt.c | 82 const int32_t kernel_y = filter_dims->h; in arm_depthwise_conv_s4_opt() local
102 const int32_t kernel_size = kernel_x * kernel_y; in arm_depthwise_conv_s4_opt()
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()
257 const uint16_t num_cols = kernel_x * kernel_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()
304 const int diff = kernel_y - ker_y_end; in arm_depthwise_conv_s4_opt()
|
| /cmsis-nn-latest/Source/PoolingFunctions/ |
| D | arm_avgpool_s8.c | 89 const int32_t kernel_y = filter_dims->h; in arm_avgpool_s8() local
112 const int32_t k_y_end = MIN(i_y * stride_y - pad_y + kernel_y, input_y); in arm_avgpool_s8()
250 const int32_t kernel_y = filter_dims->h; in arm_avgpool_s8() local
288 const int32_t kernel_y_end = MIN(kernel_y, input_y - idx_y); 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()
|
| D | arm_avgpool_s16.c | 87 const int32_t kernel_y = filter_dims->h; in arm_avgpool_s16() local
114 const int32_t k_y_end = MIN(i_y * stride_y - pad_y + kernel_y, input_y); in arm_avgpool_s16()
224 const int32_t kernel_y_end = MIN(kernel_y, input_y - 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()
|
| D | arm_max_pool_s16.c | 160 const int32_t kernel_y = filter_dims->h; in arm_max_pool_s16() local
189 const int32_t kernel_y_end = MIN(kernel_y, input_y - base_idx_y); in arm_max_pool_s16()
|
| D | arm_max_pool_s8.c | 177 const int32_t kernel_y = filter_dims->h; in arm_max_pool_s8() local
205 const int32_t kernel_y_end = MIN(kernel_y, input_y - base_idx_y); in arm_max_pool_s8()
|
