Home
last modified time | relevance | path

Searched refs:i_x (Results 1 – 3 of 3) sorted by relevance

/hal_nxp-3.5.0/mcux/mcux-sdk/CMSIS/NN/Source/PoolingFunctions/
Darm_pool_q7_HWC.c184 int16_t i_x, i_y; in arm_maxpool_q7_HWC() local
190 for (i_x = 0; i_x < dim_im_out; i_x++) in arm_maxpool_q7_HWC()
193 q7_t *target = Im_in + (i_y * dim_im_in + i_x) * ch_im_in; in arm_maxpool_q7_HWC()
196 if (i_x * stride - padding < 0) in arm_maxpool_q7_HWC()
202 win_start = Im_in + (i_y * dim_im_in + i_x * stride - padding) * ch_im_in; in arm_maxpool_q7_HWC()
205 if (i_x * stride - padding + dim_kernel >= dim_im_in) in arm_maxpool_q7_HWC()
211 … win_stop = Im_in + (i_y * dim_im_in + i_x * stride - padding + dim_kernel) * ch_im_in; in arm_maxpool_q7_HWC()
269 int16_t i_ch_in, i_x, i_y; in arm_maxpool_q7_HWC() local
276 for (i_x = 0; i_x < dim_im_out; i_x++) in arm_maxpool_q7_HWC()
281 … for (k_x = i_x * stride - padding; k_x < i_x * stride - padding + dim_kernel; k_x++) in arm_maxpool_q7_HWC()
[all …]
Darm_avgpool_s8.c97 int32_t i_x, i_y; in arm_avgpool_s8() local
102 for (i_x = 0; i_x < output_x; i_x++) in arm_avgpool_s8()
114 k_x_start = MAX(0, i_x * stride_x - pad_x); in arm_avgpool_s8()
115 k_x_end = MIN(i_x * stride_x - pad_x + kernel_x, input_x); in arm_avgpool_s8()
118 pDst = &dst[ch_src * (i_x + i_y * output_x)]; in arm_avgpool_s8()
272 for (int i_x = 0, idx_x = -pad_x; i_x < output_x; idx_x += stride_x, i_x++) in arm_avgpool_s8() local
325 int16_t i_ch_in, i_x, i_y; in arm_avgpool_s8() local
330 for (i_x = 0; i_x < output_x; i_x++) in arm_avgpool_s8()
338 … for (k_x = i_x * stride_x - pad_x; k_x < i_x * stride_x - pad_x + kernel_x; k_x++) in arm_avgpool_s8()
358 dst[i_ch_in + ch_src * (i_x + i_y * output_x)] = sum; in arm_avgpool_s8()
Darm_max_pool_s8.c188 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() local