Home
last modified time | relevance | path

Searched refs:dim_kernel (Results 1 – 8 of 8) sorted by relevance

/hal_nxp-3.5.0/mcux/mcux-sdk/CMSIS/NN/Source/ConvolutionFunctions/
Darm_convolve_HWC_q7_fast.c96 const uint16_t dim_kernel, in arm_convolve_HWC_q7_fast() argument
140 …i_ker_y = i_out_y * stride - padding; i_ker_y < i_out_y * stride - padding + dim_kernel; i_ker_y++) in arm_convolve_HWC_q7_fast()
142 …i_ker_x = i_out_x * stride - padding; i_ker_x < i_out_x * stride - padding + dim_kernel; i_ker_x++) in arm_convolve_HWC_q7_fast()
158 if (pBuffer == bufferA + 2 * ch_im_in * dim_kernel * dim_kernel) in arm_convolve_HWC_q7_fast()
161 … wt, bufferA, ch_im_out, ch_im_in * dim_kernel * dim_kernel, bias_shift, out_shift, bias, pOut); in arm_convolve_HWC_q7_fast()
176 …i_ker_y = i_out_y * stride - padding; i_ker_y < i_out_y * stride - padding + dim_kernel; i_ker_y++) in arm_convolve_HWC_q7_fast()
178 …i_ker_x = i_out_x * stride - padding; i_ker_x < i_out_x * stride - padding + dim_kernel; i_ker_x++) in arm_convolve_HWC_q7_fast()
194 if (pBuffer == bufferA + 2 * ch_im_in * dim_kernel * dim_kernel) in arm_convolve_HWC_q7_fast()
197 … wt, bufferA, ch_im_out, ch_im_in * dim_kernel * dim_kernel, bias_shift, out_shift, bias, pOut); in arm_convolve_HWC_q7_fast()
207 …i_ker_y = i_out_y * stride - padding; i_ker_y < i_out_y * stride - padding + dim_kernel; i_ker_y++) in arm_convolve_HWC_q7_fast()
[all …]
Darm_convolve_HWC_q7_basic.c79 const uint16_t dim_kernel, in arm_convolve_HWC_q7_basic() argument
108 …i_ker_y = i_out_y * stride - padding; i_ker_y < i_out_y * stride - padding + dim_kernel; i_ker_y++) in arm_convolve_HWC_q7_basic()
110 …i_ker_x = i_out_x * stride - padding; i_ker_x < i_out_x * stride - padding + dim_kernel; i_ker_x++) in arm_convolve_HWC_q7_basic()
129 if (pBuffer == bufferA + 2 * ch_im_in * dim_kernel * dim_kernel) in arm_convolve_HWC_q7_basic()
132 … wt, bufferA, ch_im_out, ch_im_in * dim_kernel * dim_kernel, bias_shift, out_shift, bias, pOut); in arm_convolve_HWC_q7_basic()
155 uint16_t colCnt = ch_im_in * dim_kernel * dim_kernel >> 2; in arm_convolve_HWC_q7_basic()
172 colCnt = ch_im_in * dim_kernel * dim_kernel & 0x3; in arm_convolve_HWC_q7_basic()
197 for (m = 0; m < dim_kernel; m++) in arm_convolve_HWC_q7_basic()
199 for (n = 0; n < dim_kernel; n++) in arm_convolve_HWC_q7_basic()
209 … wt[i * ch_im_in * dim_kernel * dim_kernel + (m * dim_kernel + n) * ch_im_in + l]; in arm_convolve_HWC_q7_basic()
Darm_convolve_HWC_q15_fast.c84 const uint16_t dim_kernel, in arm_convolve_HWC_q15_fast() argument
116 …i_ker_y = i_out_y * stride - padding; i_ker_y < i_out_y * stride - padding + dim_kernel; i_ker_y++) in arm_convolve_HWC_q15_fast()
118 …i_ker_x = i_out_x * stride - padding; i_ker_x < i_out_x * stride - padding + dim_kernel; i_ker_x++) in arm_convolve_HWC_q15_fast()
151 const q15_t *pB2 = pB + ch_im_in * dim_kernel * dim_kernel; in arm_convolve_HWC_q15_fast()
154 const q15_t *pA2 = pA + ch_im_in * dim_kernel * dim_kernel; in arm_convolve_HWC_q15_fast()
162 uint16_t colCnt = ch_im_in * dim_kernel * dim_kernel >> 1; in arm_convolve_HWC_q15_fast()
178 colCnt = ch_im_in * dim_kernel * dim_kernel & 0x1; in arm_convolve_HWC_q15_fast()
198 pA += ch_im_in * dim_kernel * dim_kernel; in arm_convolve_HWC_q15_fast()
228 for (m = 0; m < dim_kernel; m++) in arm_convolve_HWC_q15_fast()
230 for (n = 0; n < dim_kernel; n++) in arm_convolve_HWC_q15_fast()
[all …]
Darm_convolve_HWC_q7_RGB.c85 const uint16_t dim_kernel, in arm_convolve_HWC_q7_RGB() argument
118 …i_ker_y = i_out_y * stride - padding; i_ker_y < i_out_y * stride - padding + dim_kernel; i_ker_y++) in arm_convolve_HWC_q7_RGB()
120 …i_ker_x = i_out_x * stride - padding; i_ker_x < i_out_x * stride - padding + dim_kernel; i_ker_x++) in arm_convolve_HWC_q7_RGB()
179 if (pBuffer == bufferA + 2 * 3 * dim_kernel * dim_kernel) in arm_convolve_HWC_q7_RGB()
182 … wt, bufferA, ch_im_out, 3 * dim_kernel * dim_kernel, bias_shift, out_shift, bias, pOut); in arm_convolve_HWC_q7_RGB()
201 uint16_t colCnt = 3 * dim_kernel * dim_kernel >> 2; in arm_convolve_HWC_q7_RGB()
218 colCnt = 3 * dim_kernel * dim_kernel & 0x3; in arm_convolve_HWC_q7_RGB()
249 for (m = 0; m < dim_kernel; m++) in arm_convolve_HWC_q7_RGB()
251 for (n = 0; n < dim_kernel; n++) in arm_convolve_HWC_q7_RGB()
261 … wt[i * ch_im_in * dim_kernel * dim_kernel + (m * dim_kernel + n) * ch_im_in + l]; in arm_convolve_HWC_q7_RGB()
Darm_convolve_HWC_q15_basic.c79 const uint16_t dim_kernel, in arm_convolve_HWC_q15_basic() argument
108 …i_ker_y = i_out_y * stride - padding; i_ker_y < i_out_y * stride - padding + dim_kernel; i_ker_y++) in arm_convolve_HWC_q15_basic()
110 …i_ker_x = i_out_x * stride - padding; i_ker_x < i_out_x * stride - padding + dim_kernel; i_ker_x++) in arm_convolve_HWC_q15_basic()
135 uint16_t colCnt = ch_im_in * dim_kernel * dim_kernel >> 2; in arm_convolve_HWC_q15_basic()
148 colCnt = ch_im_in * dim_kernel * dim_kernel & 0x3; in arm_convolve_HWC_q15_basic()
180 for (m = 0; m < dim_kernel; m++) in arm_convolve_HWC_q15_basic()
182 for (n = 0; n < dim_kernel; n++) in arm_convolve_HWC_q15_basic()
191 … wt[i * ch_im_in * dim_kernel * dim_kernel + (m * dim_kernel + n) * ch_im_in + l]; in arm_convolve_HWC_q15_basic()
Darm_depthwise_separable_conv_HWC_q7.c87 const uint16_t dim_kernel, in arm_depthwise_separable_conv_HWC_q7() argument
122 …i_ker_y = i_out_y * stride - padding; i_ker_y < i_out_y * stride - padding + dim_kernel; i_ker_y++) in arm_depthwise_separable_conv_HWC_q7()
124 …i_ker_x = i_out_x * stride - padding; i_ker_x < i_out_x * stride - padding + dim_kernel; i_ker_x++) in arm_depthwise_separable_conv_HWC_q7()
153 uint16_t colCnt = (dim_kernel * dim_kernel) >> 1; in arm_depthwise_separable_conv_HWC_q7()
323 colCnt = (dim_kernel * dim_kernel) & 0x1; in arm_depthwise_separable_conv_HWC_q7()
352 uint16_t colCnt = (dim_kernel * dim_kernel); in arm_depthwise_separable_conv_HWC_q7()
395 for (i_ker_y = 0; i_ker_y < dim_kernel; i_ker_y++) in arm_depthwise_separable_conv_HWC_q7()
397 for (i_ker_x = 0; i_ker_x < dim_kernel; i_ker_x++) in arm_depthwise_separable_conv_HWC_q7()
404 wt[(i_ker_y * dim_kernel + i_ker_x) * ch_im_out + i_ch_out]; in arm_depthwise_separable_conv_HWC_q7()
/hal_nxp-3.5.0/mcux/mcux-sdk/CMSIS/NN/Source/PoolingFunctions/
Darm_pool_q7_HWC.c173 const uint16_t dim_kernel, in arm_maxpool_q7_HWC() argument
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()
245 if (i_y * stride - padding + dim_kernel >= dim_im_in) in arm_maxpool_q7_HWC()
251 row_end = Im_in + (i_y * stride - padding + dim_kernel) * dim_im_in * ch_im_in; in arm_maxpool_q7_HWC()
279 for (k_y = i_y * stride - padding; k_y < i_y * stride - padding + dim_kernel; k_y++) 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()
329 const uint16_t dim_kernel, in arm_avepool_q7_HWC() argument
363 if (i_x * stride - padding + dim_kernel >= dim_im_in) in arm_avepool_q7_HWC()
369 … win_stop = Im_in + (i_y * dim_im_in + i_x * stride - padding + dim_kernel) * ch_im_in; in arm_avepool_q7_HWC()
[all …]
/hal_nxp-3.5.0/mcux/mcux-sdk/CMSIS/NN/Include/
Darm_nnfunctions.h300 const uint16_t dim_kernel,
381 const uint16_t dim_kernel,
422 const uint16_t dim_kernel,
682 const uint16_t dim_kernel,
724 const uint16_t dim_kernel,
828 const uint16_t dim_kernel,
1587 const uint16_t dim_kernel,
1612 const uint16_t dim_kernel,