Searched refs:i_y (Results 1 – 3 of 3) sorted by relevance
/hal_nxp-3.5.0/mcux/mcux-sdk/CMSIS/NN/Source/PoolingFunctions/ |
D | arm_pool_q7_HWC.c | 184 int16_t i_x, i_y; in arm_maxpool_q7_HWC() local 187 for (i_y = 0; i_y < dim_im_in; i_y++) 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() 202 win_start = Im_in + (i_y * dim_im_in + i_x * stride - padding) * ch_im_in; in arm_maxpool_q7_HWC() 207 win_stop = Im_in + (i_y * dim_im_in + dim_im_in) * ch_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() 228 for (i_y = 0; i_y < dim_im_out; i_y++) in arm_maxpool_q7_HWC() 232 q7_t *target = Im_out + i_y * dim_im_out * ch_im_in; in arm_maxpool_q7_HWC() 236 if (i_y * stride - padding < 0) in arm_maxpool_q7_HWC() 242 row_start = Im_in + (i_y * stride - padding) * dim_im_in * ch_im_in; in arm_maxpool_q7_HWC() [all …]
|
D | arm_avgpool_s8.c | 97 int32_t i_x, i_y; in arm_avgpool_s8() local 100 for (i_y = 0; i_y < output_y; i_y++) in arm_avgpool_s8() 111 k_y_start = MAX(0, i_y * stride_y - pad_y); in arm_avgpool_s8() 112 k_y_end = MIN(i_y * stride_y - pad_y + kernel_y, input_y); in arm_avgpool_s8() 118 pDst = &dst[ch_src * (i_x + i_y * output_x)]; in arm_avgpool_s8() 270 for (int i_y = 0, idx_y = -pad_y; i_y < output_y; idx_y += stride_y, i_y++) in arm_avgpool_s8() local 325 int16_t i_ch_in, i_x, i_y; in arm_avgpool_s8() local 328 for (i_y = 0; i_y < output_y; i_y++) in arm_avgpool_s8() 336 for (k_y = i_y * stride_y - pad_y; k_y < i_y * stride_y - pad_y + kernel_y; k_y++) in arm_avgpool_s8() 358 dst[i_ch_in + ch_src * (i_x + i_y * output_x)] = sum; in arm_avgpool_s8()
|
D | arm_max_pool_s8.c | 186 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() local
|