Home
last modified time | relevance | path

Searched refs:stride_y (Results 1 – 24 of 24) sorted by relevance

/cmsis-nn-latest/Tests/UnitTest/
Dgenerate_test_data.py90 stride_y=1,
107 stride_y=2,
124 stride_y=1,
146 stride_y=1,
167 stride_y=2,
187 stride_y=2,
207 stride_y=3,
227 stride_y=2,
246 stride_y=1,
266 stride_y=1,
[all …]
Dpooling_settings.py38 stride_y=1, argument
63 stride_y,
94 strides=(self.stride_y, self.stride_x),
100 strides=(self.stride_y, self.stride_x),
Dconv_settings.py41 stride_y=2, argument
74 stride_y,
265 output_y = math.ceil(float(self.y_input) / float(self.stride_y))
271 …t_y = math.ceil(float(self.y_input - self.filter_y - dilation_filter_y + 1) / float(self.stride_y))
292 "stride_y": self.stride_y,
340 strides=(self.stride_y, self.stride_x),
353 … strides=(self.stride_y, self.stride_x),
367 … strides=(self.stride_y, self.stride_x),
Dtest_settings.py72 stride_y=1, argument
153 self.stride_y = stride_y
327 f.write("#define {}_STRIDE_Y {}\n".format(prefix, self.stride_y))
408 pad_along_height = max((y_output - 1) * self.stride_y + filter_y - y_input, 0)
Dmodel_extractor.py114 f.write("#define {}_STRIDE_Y {}\n".format(prefix, self.stride_y))
116 f.write("#define {}_STRIDE_H {}\n".format(prefix, self.stride_y))
193 self.stride_y = builtin_options['stride_h']
Dfully_connected_settings.py40 stride_y=1, argument
75 stride_y,
Dadd_mul_settings.py56 stride_y=1,
/cmsis-nn-latest/Tests/UnitTest/TestCases/Common/
Dconv2d_s4_weights_template.json111 "stride_h": stride_y,
Ddw_s4_weights_template.json104 "stride_h": stride_y,
/cmsis-nn-latest/Source/PoolingFunctions/
Darm_avgpool_s8.c87 const int32_t stride_y = pool_params->stride.h; in arm_avgpool_s8() local
111 const int32_t k_y_start = MAX(0, i_y * stride_y - pad_y); 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()
248 const int32_t stride_y = pool_params->stride.h; in arm_avgpool_s8() local
277 for (int i_y = 0, idx_y = -pad_y; i_y < output_y; idx_y += stride_y, i_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()
Darm_avgpool_s16.c85 const int32_t stride_y = pool_params->stride.h; in arm_avgpool_s16() local
113 const int32_t k_y_start = MAX(0, i_y * stride_y - pad_y); 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()
213 for (int i_y = 0, idx_y = -pad_y; i_y < output_y; idx_y += stride_y, i_y++) in arm_avgpool_s16()
276 for (int i_y = 0, base_idx_y = -pad_y; i_y < output_y; base_idx_y += stride_y, i_y++) in arm_avgpool_s16()
Darm_max_pool_s16.c158 const int32_t stride_y = pool_params->stride.h; in arm_max_pool_s16() local
180 for (int i_y = 0, base_idx_y = -pad_y; i_y < output_y; base_idx_y += stride_y, i_y++) in arm_max_pool_s16()
Darm_max_pool_s8.c175 const int32_t stride_y = pool_params->stride.h; in arm_max_pool_s8() local
196 for (int i_y = 0, base_idx_y = -pad_y; i_y < output_y; base_idx_y += stride_y, i_y++) in arm_max_pool_s8()
/cmsis-nn-latest/Source/ConvolutionFunctions/
Darm_transpose_conv_s8.c88 const int32_t stride_y = transpose_conv_params->stride.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()
151 for (int i_col_y = 0, i_pad_y = -pad_y; i_col_y < col_y; i_col_y++, i_pad_y += stride_y) in arm_transpose_conv_s8()
Darm_depthwise_conv_s16.c55 const int32_t stride_y, in depthwise_conv_s16_mult_4_s16() argument
65 for (int32_t in_h = -pad_y, out_h = 0, out_idx = 0; out_h < output_y; in_h += stride_y, ++out_h) in depthwise_conv_s16_mult_4_s16()
145 const uint16_t stride_y, in depthwise_conv_s16_generic_s16() argument
162 const int16_t base_idx_y = (i_out_y * stride_y) - pad_y; in depthwise_conv_s16_generic_s16()
Darm_depthwise_conv_s8.c59 const int32_t stride_y, in depthwise_conv_s8_mult_4() argument
76 for (int32_t in_h = -pad_y, out_h = 0; out_h < output_y; in_h += stride_y, ++out_h) in depthwise_conv_s8_mult_4()
167 const uint16_t stride_y, in depthwise_conv_s8_generic() argument
190 const int16_t base_idx_y = (i_out_y * stride_y) - pad_y; in depthwise_conv_s8_generic()
Darm_convolve_s4.c83 const uint16_t stride_y = conv_params->stride.h; in arm_convolve_s4() local
111 const int32_t base_idx_y = stride_y * i_out_y - pad_y; in arm_convolve_s4()
195 const int32_t base_idx_y = stride_y * i_out_y - pad_y; in arm_convolve_s4()
Darm_convolve_s16.c86 const int32_t stride_y = conv_params->stride.h; in arm_convolve_s16() local
110 const int32_t base_idx_y = stride_y * i_out_y - pad_y; in arm_convolve_s16()
Darm_convolve_s8.c83 const uint16_t stride_y = conv_params->stride.h; in arm_convolve_s8() local
135 const int32_t base_idx_y = stride_y * i_out_y - pad_y; in arm_convolve_s8()
Darm_depthwise_conv_s8_opt.c85 const int32_t stride_y = dw_conv_params->stride.h; in arm_depthwise_conv_s8_opt() local
113 … for (int i_out_y = 0, base_idx_y = -pad_y; i_out_y < output_y; base_idx_y += stride_y, i_out_y++) in arm_depthwise_conv_s8_opt()
222 const int16_t base_idx_y = (i_out_y * stride_y) - pad_y; in arm_depthwise_conv_s8_opt()
Darm_depthwise_conv_fast_s16.c92 const int32_t stride_y = dw_conv_params->stride.h; in arm_depthwise_conv_fast_s16() local
110 … for (int i_out_y = 0, base_idx_y = -pad_y; i_out_y < output_y; base_idx_y += stride_y, i_out_y++) in arm_depthwise_conv_fast_s16()
233 const int16_t base_idx_y = (i_out_y * stride_y) - pad_y; in arm_depthwise_conv_fast_s16()
Darm_depthwise_conv_s4.c56 const int32_t stride_y, in depthwise_conv_s4_generic() argument
82 const int16_t base_idx_y = (i_out_y * stride_y) - pad_y; in depthwise_conv_s4_generic()
373 const int16_t base_idx_y = (i_out_y * stride_y) - pad_y; in depthwise_conv_s4_generic()
Darm_depthwise_conv_3x3_s8.c74 const int32_t stride_y = dw_conv_params->stride.h; in arm_depthwise_conv_3x3_s8() local
95 for (int32_t in_h = -pad_y, out_h = 0, out_idx = 0; out_h < output_y; in_h += stride_y, ++out_h) in arm_depthwise_conv_3x3_s8()
Darm_depthwise_conv_s4_opt.c86 const int32_t stride_y = dw_conv_params->stride.h; in arm_depthwise_conv_s4_opt() local
114 … for (int i_out_y = 0, base_idx_y = -pad_y; i_out_y < output_y; base_idx_y += stride_y, i_out_y++) in arm_depthwise_conv_s4_opt()
264 const int16_t base_idx_y = (i_out_y * stride_y) - pad_y; in arm_depthwise_conv_s4_opt()