| /cmsis-nn-latest/Source/ConcatenationFunctions/ |
| D | arm_concatenation_s8_z.c | 51 const uint16_t input_y, in arm_concatenation_s8_z() argument
58 const uint32_t input_copy_size = input_x * input_y * input_z; in arm_concatenation_s8_z()
59 const uint32_t output_stride = input_x * input_y * output_z; in arm_concatenation_s8_z()
61 output += offset_z * (input_x * input_y); in arm_concatenation_s8_z()
|
| D | arm_concatenation_s8_w.c | 51 const uint16_t input_y, in arm_concatenation_s8_w() argument
57 const uint32_t input_copy_size = input_x * input_y * input_z * input_w; in arm_concatenation_s8_w()
59 output += offset_w * (input_x * input_y * input_z); in arm_concatenation_s8_w()
|
| D | arm_concatenation_s8_x.c | 51 const uint16_t input_y, in arm_concatenation_s8_x() argument
58 const uint32_t num_iterations = input_y * input_z * input_w; in arm_concatenation_s8_x()
|
| D | arm_concatenation_s8_y.c | 51 const uint16_t input_y, in arm_concatenation_s8_y() argument
59 const uint32_t input_copy_size = input_x * input_y; in arm_concatenation_s8_y()
|
| /cmsis-nn-latest/Tests/UnitTest/TestCases/Common/ |
| D | conv2d_s4_weights_template.json | 16 input_y,
|
| D | dw_s4_weights_template.json | 16 input_y,
|
| /cmsis-nn-latest/Source/ConvolutionFunctions/ |
| D | arm_depthwise_conv_s4.c | 46 const int32_t input_y, in depthwise_conv_s4_generic() argument
119 … const int32_t end_min_y = (input_y - base_idx_y + dilation_y - 1) / dilation_y; in depthwise_conv_s4_generic()
125 ker_y_end = MIN(kernel_y, input_y - base_idx_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()
217 ker_y_end = MIN(kernel_y, input_y - base_idx_y); in depthwise_conv_s4_generic()
303 … const int32_t end_min_y = (input_y - base_idx_y + dilation_y - 1) / dilation_y; in depthwise_conv_s4_generic()
309 ker_y_end = MIN(kernel_y, input_y - base_idx_y); in depthwise_conv_s4_generic()
364 input += (input_x * input_y * input_ch); in depthwise_conv_s4_generic()
415 … const int32_t end_min_y = (input_y - base_idx_y + dilation_y - 1) / dilation_y; in depthwise_conv_s4_generic()
421 ker_y_end = MIN(kernel_y, input_y - base_idx_y); in depthwise_conv_s4_generic()
[all …]
|
| D | arm_depthwise_conv_s16.c | 45 const int32_t input_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()
136 const uint16_t input_y, in depthwise_conv_s16_generic_s16() argument
197 … const int32_t end_min_y = (input_y - base_idx_y + dilation_y - 1) / dilation_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()
234 input += (input_x * input_y * input_ch); in depthwise_conv_s16_generic_s16()
|
| D | arm_depthwise_conv_s8.c | 49 const int32_t input_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()
157 const uint16_t input_y, in depthwise_conv_s8_generic() argument
223 … const int32_t end_min_y = (input_y - base_idx_y + dilation_y - 1) / dilation_y; in depthwise_conv_s8_generic()
229 ker_y_end = MIN(kernel_y, input_y - base_idx_y); in depthwise_conv_s8_generic()
262 input += (input_x * input_y * input_ch); in depthwise_conv_s8_generic()
|
| D | arm_convolve_s4.c | 72 const uint16_t input_y = input_dims->h; in arm_convolve_s4() local
120 if (k_y < 0 || k_y >= input_y || k_x < 0 || k_x >= input_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()
325 input_data += (input_x * input_y * input_ch); in arm_convolve_s4()
|
| D | arm_convolve_s16.c | 72 const int32_t input_y = input_dims->h; in arm_convolve_s16() local
119 if (k_y < 0 || k_y >= input_y || k_x < 0 || k_x >= input_x) in arm_convolve_s16()
282 input_data += (input_x * input_y * input_ch); in arm_convolve_s16()
|
| D | arm_depthwise_conv_3x3_s8.c | 68 const int32_t input_y = input_dims->h; in arm_depthwise_conv_3x3_s8() local
122 for (int32_t ker_h = ker_h_start; ker_h < MIN(3, input_y - in_h); ++ker_h) in arm_depthwise_conv_3x3_s8()
178 for (int32_t ker_h = ker_h_start; ker_h < MIN(3, input_y - in_h); ++ker_h) in arm_depthwise_conv_3x3_s8()
250 for (int32_t ker_h = ker_h_start; ker_h < MIN(3, input_y - in_h); ++ker_h) in arm_depthwise_conv_3x3_s8()
|
| D | arm_depthwise_conv_fast_s16.c | 85 const int32_t input_y = input_dims->h; in arm_depthwise_conv_fast_s16() local
118 if (i_ker_y < 0 || i_ker_y >= input_y || i_ker_x < 0 || i_ker_x >= input_x) in arm_depthwise_conv_fast_s16()
150 input += input_x * input_y * input_ch; 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()
427 input += (input_x * input_y * input_ch); in arm_depthwise_conv_fast_s16()
|
| D | arm_convolve_s8.c | 71 const uint16_t input_y = input_dims->h; in arm_convolve_s8() local
144 if (k_y < 0 || k_y >= input_y || k_x < 0 || k_x >= input_x) in arm_convolve_s8()
336 input_data += (input_x * input_y * input_ch); in arm_convolve_s8()
|
| D | arm_depthwise_conv_s8_opt.c | 79 const int32_t input_y = input_dims->h; in arm_depthwise_conv_s8_opt() local
121 if (i_ker_y < 0 || i_ker_y >= input_y || i_ker_x < 0 || i_ker_x >= input_x) 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()
|
| D | arm_depthwise_conv_s4_opt.c | 80 const int32_t input_y = input_dims->h; in arm_depthwise_conv_s4_opt() local
122 if (i_ker_y < 0 || i_ker_y >= input_y || i_ker_x < 0 || i_ker_x >= input_x) 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()
|
| /cmsis-nn-latest/Source/PoolingFunctions/ |
| D | arm_avgpool_s8.c | 83 const int32_t input_y = input_dims->h; in arm_avgpool_s8() local
96 const int32_t batch_input = input_x * input_y * ch_src; in arm_avgpool_s8()
112 const int32_t k_y_end = MIN(i_y * stride_y - pad_y + kernel_y, input_y); in arm_avgpool_s8()
244 const int32_t input_y = input_dims->h; in arm_avgpool_s8() local
272 const int32_t batch_size = input_x * input_y * ch_src; in arm_avgpool_s8()
288 const int32_t kernel_y_end = MIN(kernel_y, input_y - idx_y); in arm_avgpool_s8()
336 const int32_t batch_input = input_x * input_y * ch_src; in arm_avgpool_s8()
353 if (k_y >= 0 && k_x >= 0 && k_y < input_y && k_x < input_x) in arm_avgpool_s8()
|
| D | arm_avgpool_s16.c | 81 const int32_t input_y = input_dims->h; in arm_avgpool_s16() local
94 const int32_t batch_input = input_x * input_y * ch_src; in arm_avgpool_s16()
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 | 154 const int32_t input_y = input_dims->h; in arm_max_pool_s16() local
167 const int32_t batch_size = input_x * input_y * channel_in; 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()
|
| D | arm_max_pool_s8.c | 171 const int32_t input_y = input_dims->h; in arm_max_pool_s8() local
184 const int32_t batch_size = input_x * input_y * channel_in; 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()
|
| /cmsis-nn-latest/Include/ |
| D | arm_nnfunctions.h | 2239 const uint16_t input_y,
2286 const uint16_t input_y,
2333 const uint16_t input_y,
2376 const uint16_t input_y,
|
