| /cmsis-nn-3.6.0-3.5.0/Source/PoolingFunctions/ |
| D | arm_avgpool_get_buffer_sizes_s16.c | 42 int32_t arm_avgpool_s16_get_buffer_size(const int output_x, const int ch_src) in arm_avgpool_s16_get_buffer_size() argument
45 return arm_avgpool_s16_get_buffer_size_mve(output_x, ch_src); in arm_avgpool_s16_get_buffer_size()
47 return arm_avgpool_s16_get_buffer_size_dsp(output_x, ch_src); in arm_avgpool_s16_get_buffer_size()
49 (void)output_x; in arm_avgpool_s16_get_buffer_size()
55 int32_t arm_avgpool_s16_get_buffer_size_dsp(const int output_x, const int ch_src) in arm_avgpool_s16_get_buffer_size_dsp() argument
57 (void)output_x; in arm_avgpool_s16_get_buffer_size_dsp()
61 int32_t arm_avgpool_s16_get_buffer_size_mve(const int output_x, const int ch_src) in arm_avgpool_s16_get_buffer_size_mve() argument
63 (void)output_x; in arm_avgpool_s16_get_buffer_size_mve()
|
| D | arm_avgpool_get_buffer_sizes_s8.c | 42 int32_t arm_avgpool_s8_get_buffer_size(const int output_x, const int ch_src) in arm_avgpool_s8_get_buffer_size() argument
45 return arm_avgpool_s8_get_buffer_size_mve(output_x, ch_src); in arm_avgpool_s8_get_buffer_size()
47 return arm_avgpool_s8_get_buffer_size_dsp(output_x, ch_src); in arm_avgpool_s8_get_buffer_size()
49 (void)output_x; in arm_avgpool_s8_get_buffer_size()
55 int32_t arm_avgpool_s8_get_buffer_size_dsp(const int output_x, const int ch_src) in arm_avgpool_s8_get_buffer_size_dsp() argument
57 (void)output_x; in arm_avgpool_s8_get_buffer_size_dsp()
61 int32_t arm_avgpool_s8_get_buffer_size_mve(const int output_x, const int ch_src) in arm_avgpool_s8_get_buffer_size_mve() argument
63 (void)output_x; in arm_avgpool_s8_get_buffer_size_mve()
|
| D | arm_avgpool_s16.c | 84 const int32_t output_x = output_dims->w; in arm_avgpool_s16() local
98 for (int i_x = 0; i_x < output_x; i_x++) in arm_avgpool_s16()
107 int16_t *out = &dst[ch_src * (i_x + i_y * output_x)]; in arm_avgpool_s16()
194 for (int i_x = 0, idx_x = -pad_x; i_x < output_x; idx_x += stride_x, i_x++) in arm_avgpool_s16()
251 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()
285 dst[i_ch_in + ch_src * (i_x + i_y * output_x)] = sum; in arm_avgpool_s16()
|
| D | arm_avgpool_s8.c | 86 const int32_t output_x = output_dims->w; in arm_avgpool_s8() local
99 for (int i_x = 0; i_x < output_x; i_x++) in arm_avgpool_s8()
108 int8_t *out = &dst[ch_src * (i_x + i_y * output_x)]; in arm_avgpool_s8()
231 const int32_t output_x = output_dims->w; in arm_avgpool_s8() local
254 for (int i_x = 0, idx_x = -pad_x; i_x < output_x; idx_x += stride_x, i_x++) in arm_avgpool_s8()
310 for (int i_x = 0; i_x < output_x; i_x++) in arm_avgpool_s8()
338 dst[i_ch_in + ch_src * (i_x + i_y * output_x)] = sum; in arm_avgpool_s8()
|
| D | arm_max_pool_s16.c | 156 const int32_t output_x = output_dims->w; in arm_max_pool_s16() local
171 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()
205 clamp_output(dst_base, output_x * output_y * channel_in, act_min, act_max); in arm_max_pool_s16()
|
| D | arm_max_pool_s8.c | 174 const int32_t output_x = output_dims->w; in arm_max_pool_s8() local
189 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()
223 clamp_output(dst_base, output_x * output_y * channel_in, act_min, act_max); in arm_max_pool_s8()
|
| /cmsis-nn-3.6.0-3.5.0/Source/ConcatenationFunctions/ |
| D | arm_concatenation_s8_x.c | 55 const uint16_t output_x, in arm_concatenation_s8_x() argument
69 output += output_x; in arm_concatenation_s8_x()
|
| /cmsis-nn-3.6.0-3.5.0/Source/ConvolutionFunctions/ |
| D | arm_convolve_1_x_n_s8.c | 76 const uint16_t output_x = output_dims->w; in arm_convolve_1_x_n_s8() local
83 const int32_t total_pad = ((output_x - 1) * stride_x + kernel_x - input_x); in arm_convolve_1_x_n_s8()
93 const int32_t no_pad_num = MAX(output_x - (right_pad_num + left_pad_num), 0); in arm_convolve_1_x_n_s8()
94 if (right_pad_num + no_pad_num + left_pad_num != output_x) in arm_convolve_1_x_n_s8()
158 lhs_rows = output_x - out_idx; in arm_convolve_1_x_n_s8()
165 for (int i = out_idx; i < output_x; i++) in arm_convolve_1_x_n_s8()
|
| D | arm_convolve_s16.c | 72 const int32_t output_x = output_dims->w; in arm_convolve_s16() local
97 …for (int32_t base_idx_x = -pad_x, i_out_x = 0; i_out_x < output_x; base_idx_x += stride_x, i_out_x… in arm_convolve_s16()
134 … output_data[i_out_ch + (i_out_y * output_x + i_out_x) * output_ch] = (int16_t)conv_out; in arm_convolve_s16()
140 output_data += (output_x * output_y * output_ch); in arm_convolve_s16()
|
| D | arm_convolve_fast_s16.c | 81 const int32_t output_x = output_dims->w; in arm_convolve_fast_s16() local
105 for (int32_t i_out_x = 0; i_out_x < output_x; i_out_x++) in arm_convolve_fast_s16()
223 output_data += (output_x * output_y * output_ch); in arm_convolve_fast_s16()
|
| D | arm_convolve_s8.c | 76 const uint16_t output_x = output_dims->w; in arm_convolve_s8() local
114 for (int i_out_x = 0; i_out_x < output_x; i_out_x++) in arm_convolve_s8()
278 output_data += (output_x * output_y * output_ch); in arm_convolve_s8()
|
| D | arm_depthwise_conv_s16.c | 60 const int32_t output_x, in depthwise_conv_s16_mult_4_s16() argument
67 …for (int32_t in_w = -pad_x, out_w = 0, ker_h_start = MAX(0, -in_h); out_w < output_x; in_w += stri… in depthwise_conv_s16_mult_4_s16()
150 const uint16_t output_x, in depthwise_conv_s16_generic_s16() argument
163 for (int i_out_x = 0; i_out_x < output_x; i_out_x++) in depthwise_conv_s16_generic_s16()
|
| D | arm_depthwise_conv_s8.c | 64 const int32_t output_x, in depthwise_conv_s8_mult_4() argument
78 …for (int32_t in_w = -pad_x, out_w = 0, ker_h_start = MAX(0, -in_h); out_w < output_x; in_w += stri… in depthwise_conv_s8_mult_4()
172 const uint16_t output_x, in depthwise_conv_s8_generic() argument
191 for (int i_out_x = 0; i_out_x < output_x; i_out_x++) in depthwise_conv_s8_generic()
|
| D | arm_depthwise_conv_s8_opt.c | 88 const int32_t output_x = output_dims->w; in arm_depthwise_conv_s8_opt() local
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()
223 for (int i_out_x = 0; i_out_x < output_x; i_out_x++) in arm_depthwise_conv_s8_opt()
|
| D | arm_depthwise_conv_fast_s16.c | 95 const int32_t output_x = output_dims->w; in arm_depthwise_conv_fast_s16() local
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()
234 for (int i_out_x = 0; i_out_x < output_x; i_out_x++) in arm_depthwise_conv_fast_s16()
|
| D | arm_depthwise_conv_3x3_s8.c | 77 const int32_t output_x = output_dims->w; in arm_depthwise_conv_3x3_s8() local
97 …for (int32_t in_w = -pad_x, out_w = 0, ker_h_start = MAX(0, -in_h); out_w < output_x; in_w += stri… in arm_depthwise_conv_3x3_s8()
|
| /cmsis-nn-3.6.0-3.5.0/Include/ |
| D | arm_nnfunctions.h | 1708 const uint16_t output_x,
|
| /cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/ |
| D | generate_test_data.py | 409 def set_output_dims_and_padding(self, output_x, output_y): argument
410 self.x_output = output_x
|
