Home
last modified time | relevance | path

Searched refs:k_x (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
137 for (k_x = k_x_start; k_x < k_x_end; k_x++) in arm_avgpool_s8()
139 pTmpInner = pTmp + (ch_src * (k_x + k_y * input_x)); in arm_avgpool_s8()
222 for (k_x = k_x_start; k_x < k_x_end; k_x++) in arm_avgpool_s8()
224 sum += pTmp[ch_src * (k_x + k_y * input_x)]; in arm_avgpool_s8()
288 for (int k_x = kernel_x_start; k_x < kernel_x_end; k_x++) 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
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()
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
281 … for (k_x = i_x * stride - padding; k_x < i_x * stride - padding + dim_kernel; k_x++) 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()
445 … for (k_x = i_x * stride - padding; k_x < i_x * stride - padding + dim_kernel; k_x++) 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.c202 for (int k_x = ker_x_start; k_x < kernel_x_end; k_x++) 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()