| /cmsis-nn-latest/Source/ConvolutionFunctions/ |
| D | arm_convolve_1_x_n_s4.c | 79 const uint16_t stride_x = conv_params->stride.w; in arm_convolve_1_x_n_s4() local
82 const int32_t total_pad = ((output_x - 1) * stride_x + kernel_x - input_x); in arm_convolve_1_x_n_s4()
90 …t right_pad_num = pad_x + asym_pad != 0 ? MAX(1, (pad_x + asym_pad + stride_x - 1) / stride_x) : 0; in arm_convolve_1_x_n_s4()
91 const int32_t left_pad_num = pad_x != 0 ? MAX(1, (pad_x + stride_x - 1) / stride_x) : 0; in arm_convolve_1_x_n_s4()
105 const int32_t lhs_offset = input_ch * stride_x; in arm_convolve_1_x_n_s4()
111 const int32_t est_input_x_idx = stride_x * i - pad_x; in arm_convolve_1_x_n_s4()
127 int32_t input_start = stride_x * lhs_rows - pad_x; in arm_convolve_1_x_n_s4()
163 const int32_t est_input_x_idx = stride_x * i - pad_x; in arm_convolve_1_x_n_s4()
|
| D | arm_convolve_1_x_n_s8.c | 77 const int32_t stride_x = conv_params->stride.w; in arm_convolve_1_x_n_s8() local
80 const int32_t total_pad = ((output_x - 1) * stride_x + kernel_x - input_x); in arm_convolve_1_x_n_s8()
88 …t right_pad_num = pad_x + asym_pad != 0 ? MAX(1, (pad_x + asym_pad + stride_x - 1) / stride_x) : 0; in arm_convolve_1_x_n_s8()
89 const int32_t left_pad_num = pad_x != 0 ? MAX(1, (pad_x + stride_x - 1) / stride_x) : 0; in arm_convolve_1_x_n_s8()
97 const int32_t lhs_offset = input_ch * stride_x; in arm_convolve_1_x_n_s8()
147 int32_t input_start = stride_x * lhs_rows - pad_x; in arm_convolve_1_x_n_s8()
185 input_start = (stride_x * (left_pad_num + no_pad_num) - pad_x) * input_ch; in arm_convolve_1_x_n_s8()
|
| D | arm_convolve_get_buffer_sizes_s4.c | 64 const int32_t stride_x = conv_params->stride.w; in arm_convolve_1_x_n_s4_get_buffer_size_mve() local
65 const int32_t total_pad = ((output_x - 1) * stride_x + kernel_x - input_x); in arm_convolve_1_x_n_s4_get_buffer_size_mve()
68 …t right_pad_num = pad_x + asym_pad != 0 ? MAX(1, (pad_x + asym_pad + stride_x - 1) / stride_x) : 0; in arm_convolve_1_x_n_s4_get_buffer_size_mve()
69 const int32_t left_pad_num = pad_x != 0 ? MAX(1, (pad_x + stride_x - 1) / stride_x) : 0; in arm_convolve_1_x_n_s4_get_buffer_size_mve()
|
| D | arm_convolve_get_buffer_sizes_s8.c | 64 const int32_t stride_x = conv_params->stride.w; in arm_convolve_1_x_n_s8_get_buffer_size_mve() local
65 const int32_t total_pad = ((output_x - 1) * stride_x + kernel_x - input_x); in arm_convolve_1_x_n_s8_get_buffer_size_mve()
68 …t right_pad_num = pad_x + asym_pad != 0 ? MAX(1, (pad_x + asym_pad + stride_x - 1) / stride_x) : 0; in arm_convolve_1_x_n_s8_get_buffer_size_mve()
69 const int32_t left_pad_num = pad_x != 0 ? MAX(1, (pad_x + stride_x - 1) / stride_x) : 0; in arm_convolve_1_x_n_s8_get_buffer_size_mve()
|
| D | arm_transpose_conv_s8.c | 87 const int32_t stride_x = transpose_conv_params->stride.w; in arm_transpose_conv_s8() local
148 const int32_t col_x = (output_x + pad_x_offset + pad_x - kernel_x) / stride_x + 1; in arm_transpose_conv_s8()
153 for (int i_col_x = 0, i_pad_x = -pad_x; i_col_x < col_x; i_col_x++, i_pad_x += stride_x) in arm_transpose_conv_s8()
|
| D | arm_depthwise_conv_s16.c | 54 const int32_t stride_x, in depthwise_conv_s16_mult_4_s16() argument
67 …in_w = -pad_x, out_w = 0, ker_h_start = MAX(0, -in_h); out_w < output_x; in_w += stride_x, ++out_w) in depthwise_conv_s16_mult_4_s16()
144 const uint16_t stride_x, in depthwise_conv_s16_generic_s16() argument
165 const int16_t base_idx_x = (i_out_x * stride_x) - pad_x; in depthwise_conv_s16_generic_s16()
|
| D | arm_depthwise_conv_s8.c | 58 const int32_t stride_x, in depthwise_conv_s8_mult_4() argument
78 …in_w = -pad_x, out_w = 0, ker_h_start = MAX(0, -in_h); out_w < output_x; in_w += stride_x, ++out_w) in depthwise_conv_s8_mult_4()
166 const uint16_t stride_x, in depthwise_conv_s8_generic() argument
193 const int16_t base_idx_x = (i_out_x * stride_x) - pad_x; in depthwise_conv_s8_generic()
|
| D | arm_convolve_s4.c | 82 const uint16_t stride_x = conv_params->stride.w; in arm_convolve_s4() local
110 const int32_t base_idx_x = stride_x * i_out_x - pad_x; in arm_convolve_s4()
194 const int32_t base_idx_x = stride_x * i_out_x - pad_x; in arm_convolve_s4()
|
| D | arm_convolve_s16.c | 85 const int32_t stride_x = conv_params->stride.w; in arm_convolve_s16() local
109 const int32_t base_idx_x = stride_x * i_out_x - pad_x; in arm_convolve_s16()
|
| D | arm_convolve_s8.c | 82 const uint16_t stride_x = conv_params->stride.w; in arm_convolve_s8() local
134 const int32_t base_idx_x = stride_x * i_out_x - pad_x; in arm_convolve_s8()
|
| D | arm_depthwise_conv_s8_opt.c | 84 const int32_t stride_x = dw_conv_params->stride.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()
225 const int16_t base_idx_x = (i_out_x * stride_x) - pad_x; in arm_depthwise_conv_s8_opt()
|
| D | arm_depthwise_conv_fast_s16.c | 91 const int32_t stride_x = dw_conv_params->stride.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()
236 const int16_t base_idx_x = (i_out_x * stride_x) - pad_x; in arm_depthwise_conv_fast_s16()
|
| D | arm_depthwise_conv_s4.c | 55 const int32_t stride_x, in depthwise_conv_s4_generic() argument
85 const int16_t base_idx_x = (i_out_x * stride_x) - pad_x; in depthwise_conv_s4_generic()
376 const int16_t base_idx_x = (i_out_x * stride_x) - pad_x; in depthwise_conv_s4_generic()
|
| /cmsis-nn-latest/Tests/UnitTest/ |
| D | generate_test_data.py | 89 stride_x=1,
106 stride_x=2,
123 stride_x=1,
145 stride_x=3,
166 stride_x=2,
186 stride_x=2,
206 stride_x=3,
226 stride_x=1,
245 stride_x=1,
265 stride_x=2,
[all …]
|
| D | pooling_settings.py | 37 stride_x=1, argument
62 stride_x,
94 strides=(self.stride_y, self.stride_x),
100 strides=(self.stride_y, self.stride_x),
|
| D | conv_settings.py | 40 stride_x=2, argument
73 stride_x,
264 output_x = math.ceil(float(self.x_input) / float(self.stride_x))
270 …t_x = math.ceil(float(self.x_input - self.filter_x - dilation_filter_x + 1) / float(self.stride_x))
291 "stride_x": self.stride_x,
340 strides=(self.stride_y, self.stride_x),
353 … strides=(self.stride_y, self.stride_x),
367 … strides=(self.stride_y, self.stride_x),
|
| D | test_settings.py | 71 stride_x=1, argument
152 self.stride_x = stride_x
326 f.write("#define {}_STRIDE_X {}\n".format(prefix, self.stride_x))
407 pad_along_width = max((x_output - 1) * self.stride_x + filter_x - x_input, 0)
|
| D | model_extractor.py | 113 f.write("#define {}_STRIDE_X {}\n".format(prefix, self.stride_x))
115 f.write("#define {}_STRIDE_W {}\n".format(prefix, self.stride_x))
191 self.stride_x = builtin_options['stride_w']
|
| D | fully_connected_settings.py | 39 stride_x=1, argument
74 stride_x,
|
| /cmsis-nn-latest/Tests/UnitTest/TestCases/Common/ |
| D | conv2d_s4_weights_template.json | 110 "stride_w": stride_x,
|
| D | dw_s4_weights_template.json | 103 "stride_w": stride_x,
|
| /cmsis-nn-latest/Source/PoolingFunctions/ |
| D | arm_avgpool_s8.c | 88 const int32_t stride_x = pool_params->stride.w; in arm_avgpool_s8() local
114 const int32_t k_x_start = MAX(0, i_x * stride_x - pad_x); in arm_avgpool_s8()
115 const int32_t k_x_end = MIN(i_x * stride_x - pad_x + kernel_x, input_x); in arm_avgpool_s8()
249 const int32_t stride_x = pool_params->stride.w; in arm_avgpool_s8() local
279 for (int i_x = 0, idx_x = -pad_x; i_x < output_x; idx_x += stride_x, i_x++) in arm_avgpool_s8()
351 … for (int k_x = i_x * stride_x - pad_x; k_x < i_x * stride_x - pad_x + kernel_x; k_x++) in arm_avgpool_s8()
|
| D | arm_avgpool_s16.c | 86 const int32_t stride_x = pool_params->stride.w; in arm_avgpool_s16() local
116 const int32_t k_x_start = MAX(0, i_x * stride_x - pad_x); in arm_avgpool_s16()
117 const int32_t k_x_end = MIN(i_x * stride_x - pad_x + kernel_x, input_x); in arm_avgpool_s16()
215 for (int i_x = 0, idx_x = -pad_x; i_x < output_x; idx_x += stride_x, i_x++) in arm_avgpool_s16()
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()
|
| D | arm_max_pool_s16.c | 159 const int32_t stride_x = pool_params->stride.w; in arm_max_pool_s16() local
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()
|
| D | arm_max_pool_s8.c | 176 const int32_t stride_x = pool_params->stride.w; in arm_max_pool_s8() local
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()
|
