Home
last modified time | relevance | path

Searched refs:output_y (Results 1 – 20 of 20) sorted by relevance

/cmsis-nn-latest/Source/ConcatenationFunctions/
Darm_concatenation_s8_y.c55 const uint16_t output_y, in arm_concatenation_s8_y() argument
60 const uint32_t output_stride = input_x * output_y; in arm_concatenation_s8_y()
/cmsis-nn-latest/Tests/UnitTest/TestCases/Common/
Dconv2d_s4_weights_template.json70 output_y,
Ddw_s4_weights_template.json65 output_y,
/cmsis-nn-latest/Source/ConvolutionFunctions/
Darm_transpose_conv_s8.c79 const int32_t output_y = output_dims->h; in arm_transpose_conv_s8() local
112 … arm_memset_s8((int8_t *)img_buf_ptr, 0, output_x * output_y * output_ch * sizeof(int32_t)); in arm_transpose_conv_s8()
118 for (int i = 0; i < output_x * output_y; i++) in arm_transpose_conv_s8()
147 const int32_t col_y = (output_y + pad_y_offset + pad_y - kernel_y) / stride_y + 1; in arm_transpose_conv_s8()
161 … if (i_ker_y >= 0 && i_ker_y < output_y && i_ker_x >= 0 && i_ker_x < output_x) in arm_transpose_conv_s8()
176 for (int i = 0; i < output_x * output_y; i++) in arm_transpose_conv_s8()
Darm_convolve_s4.c77 const uint16_t output_y = output_dims->h; in arm_convolve_s4() local
106 for (int i_out_y = 0; i_out_y < output_y; i_out_y++) in arm_convolve_s4()
190 for (int i_out_y = 0; i_out_y < output_y; i_out_y++) in arm_convolve_s4()
326 output_data += (output_x * output_y * output_ch); in arm_convolve_s4()
Darm_convolve_s16.c77 const int32_t output_y = output_dims->h; in arm_convolve_s16() local
105 for (int32_t i_out_y = 0; i_out_y < output_y; i_out_y++) in arm_convolve_s16()
283 output_data += (output_x * output_y * output_ch); in arm_convolve_s16()
Darm_depthwise_conv_s16.c61 const int32_t output_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()
151 const uint16_t output_y, in depthwise_conv_s16_generic_s16() argument
160 for (int i_out_y = 0; i_out_y < output_y; i_out_y++) in depthwise_conv_s16_generic_s16()
Darm_depthwise_conv_s8.c65 const int32_t output_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()
173 const uint16_t output_y, in depthwise_conv_s8_generic() argument
188 for (int i_out_y = 0; i_out_y < output_y; i_out_y++) in depthwise_conv_s8_generic()
Darm_convolve_s8.c77 const uint16_t output_y = output_dims->h; in arm_convolve_s8() local
130 for (int i_out_y = 0; i_out_y < output_y; i_out_y++) in arm_convolve_s8()
337 output_data += (output_x * output_y * output_ch); in arm_convolve_s8()
Darm_depthwise_conv_s8_opt.c89 const int32_t output_y = output_dims->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()
220 for (int i_out_y = 0; i_out_y < output_y; i_out_y++) in arm_depthwise_conv_s8_opt()
Darm_depthwise_conv_fast_s16.c96 const int32_t output_y = output_dims->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()
231 for (int i_out_y = 0; i_out_y < output_y; i_out_y++) in arm_depthwise_conv_fast_s16()
Darm_depthwise_conv_s4.c62 const int32_t output_y, in depthwise_conv_s4_generic() argument
80 for (int i_out_y = 0; i_out_y < output_y; i_out_y++) in depthwise_conv_s4_generic()
371 for (int i_out_y = 0; i_out_y < output_y; i_out_y++) in depthwise_conv_s4_generic()
Darm_depthwise_conv_3x3_s8.c78 const int32_t output_y = output_dims->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.c90 const int32_t output_y = output_dims->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()
262 for (int i_out_y = 0; i_out_y < output_y; i_out_y++) in arm_depthwise_conv_s4_opt()
/cmsis-nn-latest/Source/PoolingFunctions/
Darm_avgpool_s16.c83 const int32_t output_y = output_dims->h; in arm_avgpool_s16() local
105 const int32_t batch_output = output_x * output_y * ch_src; in arm_avgpool_s16()
109 for (int i_y = 0; i_y < output_y; i_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()
271 const int32_t batch_output = output_x * output_y * ch_src; 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_avgpool_s8.c85 const int32_t output_y = output_dims->h; in arm_avgpool_s8() local
97 const int32_t batch_output = output_x * output_y * ch_src; in arm_avgpool_s8()
107 for (int i_y = 0; i_y < output_y; i_y++) in arm_avgpool_s8()
246 const int32_t output_y = output_dims->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()
337 const int32_t batch_output = output_x * output_y * ch_src; in arm_avgpool_s8()
341 for (int i_y = 0; i_y < output_y; i_y++) in arm_avgpool_s8()
Darm_max_pool_s16.c156 const int32_t output_y = output_dims->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()
216 clamp_output(dst_base, output_x * output_y * channel_in, act_min, act_max); in arm_max_pool_s16()
Darm_max_pool_s8.c173 const int32_t output_y = output_dims->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()
232 clamp_output(dst_base, output_x * output_y * channel_in, act_min, act_max); in arm_max_pool_s8()
/cmsis-nn-latest/Tests/UnitTest/
Dconv_settings.py265 output_y = math.ceil(float(self.y_input) / float(self.stride_y))
271output_y = math.ceil(float(self.y_input - self.filter_y - dilation_filter_y + 1) / float(self.stri…
282 "output_y": output_y,
/cmsis-nn-latest/Include/
Darm_nnfunctions.h2290 const uint16_t output_y,