Home
last modified time | relevance | path

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

/hal_nxp-3.5.0/mcux/mcux-sdk/CMSIS/NN/Source/PoolingFunctions/
Darm_avgpool_s8.c98 int32_t k_x, k_y; in arm_avgpool_s8() local
135 for (k_y = k_y_start; k_y < k_y_end; k_y++) in arm_avgpool_s8()
139 pTmpInner = pTmp + (ch_src * (k_x + k_y * input_x)); in arm_avgpool_s8()
220 for (k_y = k_y_start; k_y < k_y_end; k_y++) in arm_avgpool_s8()
224 sum += pTmp[ch_src * (k_x + k_y * input_x)]; in arm_avgpool_s8()
286 for (int k_y = kernel_y_start; k_y < kernel_y_end; k_y++) in arm_avgpool_s8() local
290 const q7_t *start = src + ch_src * (k_x + idx_x + (k_y + idx_y) * input_x); in arm_avgpool_s8()
326 int16_t k_x, k_y; in arm_avgpool_s8() local
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()
340 if (k_y >= 0 && k_x >= 0 && k_y < input_y && k_x < input_x) in arm_avgpool_s8()
[all …]
Darm_pool_q7_HWC.c270 int16_t k_x, k_y; in arm_maxpool_q7_HWC() local
279 for (k_y = i_y * stride - padding; k_y < i_y * stride - padding + dim_kernel; k_y++) in arm_maxpool_q7_HWC()
283 if (k_y >= 0 && k_x >= 0 && k_y < dim_im_in && k_x < dim_im_in) in arm_maxpool_q7_HWC()
285 if (Im_in[i_ch_in + ch_im_in * (k_x + k_y * dim_im_in)] > max) in arm_maxpool_q7_HWC()
287 max = Im_in[i_ch_in + ch_im_in * (k_x + k_y * dim_im_in)]; in arm_maxpool_q7_HWC()
433 int16_t k_x, k_y; in arm_avepool_q7_HWC()
443 for (k_y = i_y * stride - padding; k_y < i_y * stride - padding + dim_kernel; k_y++) in arm_avepool_q7_HWC()
447 if (k_y >= 0 && k_x >= 0 && k_y < dim_im_in && k_x < dim_im_in) in arm_avepool_q7_HWC()
449 sum += Im_in[i_ch_in + ch_im_in * (k_x + k_y * dim_im_in)]; in arm_avepool_q7_HWC()
Darm_max_pool_s8.c200 for (int k_y = ker_y_start; k_y < kernel_y_end; k_y++) in arm_max_pool_s8() local
204 … const q7_t *start = src + channel_in * (k_x + base_idx_x + (k_y + base_idx_y) * input_x); in arm_max_pool_s8()