| /cmsis-nn-3.6.0-3.5.0/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-3.6.0-3.5.0/Source/ConvolutionFunctions/ |
| 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_convolve_s16.c | 68 const int32_t input_y = input_dims->h; in arm_convolve_s16() local
105 const int32_t end_min_y = (input_y - base_idx_y + dilation_y - 1) / dilation_y; in arm_convolve_s16()
139 input_data += (input_x * input_y * input_ch); in arm_convolve_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_fast_s16.c | 77 const int32_t input_y = input_dims->h; in arm_convolve_fast_s16() local
113 if (i_ker_y < 0 || i_ker_y >= input_y || i_ker_x < 0 || i_ker_x >= input_x) in arm_convolve_fast_s16()
222 input_data += (input_x * input_y * input_ch); in arm_convolve_fast_s16()
|
| D | arm_convolve_s8.c | 72 const uint16_t input_y = input_dims->h; in arm_convolve_s8() local
126 if (k_y < 0 || k_y >= input_y || k_x < 0 || k_x >= input_x) in arm_convolve_s8()
277 input_data += (input_x * input_y * input_ch); in arm_convolve_s8()
|
| 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_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()
|
| /cmsis-nn-3.6.0-3.5.0/Source/PoolingFunctions/ |
| D | arm_avgpool_s16.c | 81 const int32_t input_y = input_dims->h; in arm_avgpool_s16() local
101 const int32_t k_y_end = MIN(i_y * stride_y - pad_y + kernel_y, input_y); in arm_avgpool_s16()
203 const int32_t kernel_y_end = MIN(kernel_y, input_y - idx_y); in arm_avgpool_s16()
258 const int32_t kernel_y_end = MIN(kernel_y, input_y - base_idx_y); in arm_avgpool_s16()
|
| D | arm_avgpool_s8.c | 83 const int32_t input_y = input_dims->h; in arm_avgpool_s8() local
102 const int32_t k_y_end = MIN(i_y * stride_y - pad_y + kernel_y, input_y); in arm_avgpool_s8()
228 const int32_t input_y = input_dims->h; in arm_avgpool_s8() local
263 const int32_t kernel_y_end = MIN(kernel_y, input_y - idx_y); in arm_avgpool_s8()
320 if (k_y >= 0 && k_x >= 0 && k_y < input_y && k_x < input_x) in arm_avgpool_s8()
|
| D | arm_max_pool_s16.c | 153 const int32_t input_y = input_dims->h; in arm_max_pool_s16() local
178 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
196 const int32_t kernel_y_end = MIN(kernel_y, input_y - base_idx_y); in arm_max_pool_s8()
|
| /cmsis-nn-3.6.0-3.5.0/Include/ |
| D | arm_nnfunctions.h | 1704 const uint16_t input_y,
1751 const uint16_t input_y,
1798 const uint16_t input_y,
1841 const uint16_t input_y,
|
