| /hal_nxp-3.5.0/mcux/mcux-sdk/CMSIS/NN/Source/PoolingFunctions/ |
| D | arm_pool_q7_HWC.c | 172 const uint16_t ch_im_in, in arm_maxpool_q7_HWC() argument
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()
216 memmove(target, win_start, ch_im_in); in arm_maxpool_q7_HWC()
219 win_start += ch_im_in; in arm_maxpool_q7_HWC()
220 for (; win_start < win_stop; win_start += ch_im_in) in arm_maxpool_q7_HWC()
222 compare_and_replace_if_larger_q7(target, win_start, ch_im_in); in arm_maxpool_q7_HWC()
232 q7_t *target = Im_out + i_y * dim_im_out * ch_im_in; in arm_maxpool_q7_HWC()
[all …]
|
| /hal_nxp-3.5.0/mcux/mcux-sdk/CMSIS/NN/Source/ConvolutionFunctions/ |
| D | arm_convolve_HWC_q7_fast.c | 93 const uint16_t ch_im_in, in arm_convolve_HWC_q7_fast() argument
121 if (ch_im_in % 4 != 0 || ch_im_out % 2 != 0) in arm_convolve_HWC_q7_fast()
147 memset(pBuffer, 0, sizeof(q15_t) * ch_im_in); in arm_convolve_HWC_q7_fast()
152 … (q7_t *)Im_in + (i_ker_y * dim_im_in + i_ker_x) * ch_im_in, pBuffer, ch_im_in); in arm_convolve_HWC_q7_fast()
154 pBuffer += ch_im_in; 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()
183 memset(pBuffer, 0, sizeof(q15_t) * ch_im_in); in arm_convolve_HWC_q7_fast()
188 … (q7_t *)Im_in + (i_ker_y * dim_im_in + i_ker_x) * ch_im_in, pBuffer, ch_im_in); in arm_convolve_HWC_q7_fast()
190 pBuffer += ch_im_in; in arm_convolve_HWC_q7_fast()
[all …]
|
| D | arm_convolve_HWC_q7_fast_nonsquare.c | 77 const uint16_t ch_im_in, in arm_convolve_HWC_q7_fast_nonsquare() argument
109 if (ch_im_in % 4 != 0 || ch_im_out % 2 != 0) in arm_convolve_HWC_q7_fast_nonsquare()
137 memset(pBuffer, 0, sizeof(q15_t) * ch_im_in); in arm_convolve_HWC_q7_fast_nonsquare()
142 … (q7_t *)Im_in + (i_ker_y * dim_im_in_x + i_ker_x) * ch_im_in, pBuffer, ch_im_in); in arm_convolve_HWC_q7_fast_nonsquare()
144 pBuffer += ch_im_in; in arm_convolve_HWC_q7_fast_nonsquare()
148 if (pBuffer == bufferA + 2 * ch_im_in * dim_kernel_x * dim_kernel_y) in arm_convolve_HWC_q7_fast_nonsquare()
151 …wt, bufferA, ch_im_out, ch_im_in * dim_kernel_x * dim_kernel_y, bias_shift, out_shift, bias, pOut); in arm_convolve_HWC_q7_fast_nonsquare()
175 memset(pBuffer, 0, sizeof(q15_t) * ch_im_in); in arm_convolve_HWC_q7_fast_nonsquare()
180 … (q7_t *)Im_in + (i_ker_y * dim_im_in_x + i_ker_x) * ch_im_in, pBuffer, ch_im_in); in arm_convolve_HWC_q7_fast_nonsquare()
182 pBuffer += ch_im_in; in arm_convolve_HWC_q7_fast_nonsquare()
[all …]
|
| D | arm_depthwise_separable_conv_HWC_q7.c | 84 const uint16_t ch_im_in, in arm_depthwise_separable_conv_HWC_q7() argument
112 if (ch_im_in != ch_im_out) in arm_depthwise_separable_conv_HWC_q7()
129 memset(pBuffer, 0, ch_im_in); in arm_depthwise_separable_conv_HWC_q7()
135 … memcpy(pBuffer, (q7_t *)Im_in + (i_ker_y * dim_im_in + i_ker_x) * ch_im_in, ch_im_in); in arm_depthwise_separable_conv_HWC_q7()
137 pBuffer += ch_im_in; in arm_depthwise_separable_conv_HWC_q7()
167 pB += ch_im_in; in arm_depthwise_separable_conv_HWC_q7()
169 pB += ch_im_in; in arm_depthwise_separable_conv_HWC_q7()
173 pA += ch_im_in; in arm_depthwise_separable_conv_HWC_q7()
175 pA += ch_im_in; in arm_depthwise_separable_conv_HWC_q7()
199 pB += ch_im_in; in arm_depthwise_separable_conv_HWC_q7()
[all …]
|
| D | arm_depthwise_separable_conv_HWC_q7_nonsquare.c | 77 const uint16_t ch_im_in, in arm_depthwise_separable_conv_HWC_q7_nonsquare() argument
120 if (ch_im_in != ch_im_out) in arm_depthwise_separable_conv_HWC_q7_nonsquare()
139 memset(pBuffer, 0, ch_im_in); in arm_depthwise_separable_conv_HWC_q7_nonsquare()
145 … memcpy(pBuffer, (q7_t *)Im_in + (i_ker_y * dim_im_in_x + i_ker_x) * ch_im_in, ch_im_in); in arm_depthwise_separable_conv_HWC_q7_nonsquare()
147 pBuffer += ch_im_in; in arm_depthwise_separable_conv_HWC_q7_nonsquare()
177 pB += ch_im_in; in arm_depthwise_separable_conv_HWC_q7_nonsquare()
179 pB += ch_im_in; in arm_depthwise_separable_conv_HWC_q7_nonsquare()
183 pA += ch_im_in; in arm_depthwise_separable_conv_HWC_q7_nonsquare()
185 pA += ch_im_in; in arm_depthwise_separable_conv_HWC_q7_nonsquare()
209 pB += ch_im_in; in arm_depthwise_separable_conv_HWC_q7_nonsquare()
[all …]
|
| D | arm_convolve_1x1_HWC_q7_fast_nonsquare.c | 84 const uint16_t ch_im_in, in arm_convolve_1x1_HWC_q7_fast_nonsquare() argument
117 …if (ch_im_in % 4 != 0 || ch_im_out % 2 != 0 || dim_kernel_x != 1 || dim_kernel_y != 1 || padding_x… in arm_convolve_1x1_HWC_q7_fast_nonsquare()
130 (q7_t *)Im_in + (i_out_y * dim_im_in_x + i_out_x) * ch_im_in, pBuffer, ch_im_in); in arm_convolve_1x1_HWC_q7_fast_nonsquare()
131 pBuffer += ch_im_in; in arm_convolve_1x1_HWC_q7_fast_nonsquare()
133 if (pBuffer == bufferA + 2 * ch_im_in * dim_kernel_x * dim_kernel_y) in arm_convolve_1x1_HWC_q7_fast_nonsquare()
136 wt, bufferA, ch_im_out, ch_im_in, bias_shift, out_shift, bias, pOut); in arm_convolve_1x1_HWC_q7_fast_nonsquare()
152 uint16_t colCnt = ch_im_in * dim_kernel_x * dim_kernel_y >> 2; in arm_convolve_1x1_HWC_q7_fast_nonsquare()
170 colCnt = ch_im_in * dim_kernel_y * dim_kernel_x & 0x3; in arm_convolve_1x1_HWC_q7_fast_nonsquare()
190 …if (ch_im_in % 4 != 0 || ch_im_out % 2 != 0 || dim_kernel_x != 1 || dim_kernel_y != 1 || padding_x… in arm_convolve_1x1_HWC_q7_fast_nonsquare()
213 for (l = 0; l < ch_im_in; l++) in arm_convolve_1x1_HWC_q7_fast_nonsquare()
[all …]
|
| D | arm_convolve_HWC_q15_fast.c | 81 const uint16_t ch_im_in, in arm_convolve_HWC_q15_fast() argument
103 if (ch_im_in % 2 != 0 || ch_im_out % 2 != 0) in arm_convolve_HWC_q15_fast()
123 memset(pBuffer, 0, sizeof(q15_t) * ch_im_in); in arm_convolve_HWC_q15_fast()
130 (q15_t *)Im_in + (i_ker_y * dim_im_in + i_ker_x) * ch_im_in, in arm_convolve_HWC_q15_fast()
131 sizeof(q15_t) * ch_im_in); in arm_convolve_HWC_q15_fast()
133 pBuffer += ch_im_in; 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()
[all …]
|
| D | arm_convolve_HWC_q15_fast_nonsquare.c | 87 const uint16_t ch_im_in, in arm_convolve_HWC_q15_fast_nonsquare() argument
113 if (ch_im_in % 2 != 0 || ch_im_out % 2 != 0) in arm_convolve_HWC_q15_fast_nonsquare()
135 memset(pBuffer, 0, sizeof(q15_t) * ch_im_in); in arm_convolve_HWC_q15_fast_nonsquare()
142 (q15_t *)Im_in + (i_ker_y * dim_im_in_x + i_ker_x) * ch_im_in, in arm_convolve_HWC_q15_fast_nonsquare()
143 sizeof(q15_t) * ch_im_in); in arm_convolve_HWC_q15_fast_nonsquare()
145 pBuffer += ch_im_in; in arm_convolve_HWC_q15_fast_nonsquare()
163 const q15_t *pB2 = pB + ch_im_in * dim_kernel_y * dim_kernel_x; in arm_convolve_HWC_q15_fast_nonsquare()
166 const q15_t *pA2 = pA + ch_im_in * dim_kernel_y * dim_kernel_x; in arm_convolve_HWC_q15_fast_nonsquare()
174 uint16_t colCnt = ch_im_in * dim_kernel_y * dim_kernel_x >> 1; in arm_convolve_HWC_q15_fast_nonsquare()
190 colCnt = ch_im_in * dim_kernel_y * dim_kernel_x & 0x1; in arm_convolve_HWC_q15_fast_nonsquare()
[all …]
|
| D | arm_convolve_HWC_q7_basic.c | 76 const uint16_t ch_im_in, in arm_convolve_HWC_q7_basic() argument
116 memset(pBuffer, 0, sizeof(q15_t) * ch_im_in); in arm_convolve_HWC_q7_basic()
122 … (q7_t *)Im_in + (i_ker_y * dim_im_in + i_ker_x) * ch_im_in, pBuffer, ch_im_in); in arm_convolve_HWC_q7_basic()
124 pBuffer += ch_im_in; 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()
206 for (l = 0; l < ch_im_in; l++) in arm_convolve_HWC_q7_basic()
208 conv_out += Im_in[(in_row * dim_im_in + in_col) * ch_im_in + l] * in arm_convolve_HWC_q7_basic()
[all …]
|
| D | arm_convolve_HWC_q7_basic_nonsquare.c | 71 const uint16_t ch_im_in, in arm_convolve_HWC_q7_basic_nonsquare() argument
117 memset(pBuffer, 0, sizeof(q15_t) * ch_im_in); in arm_convolve_HWC_q7_basic_nonsquare()
123 … (q7_t *)Im_in + (i_ker_y * dim_im_in_x + i_ker_x) * ch_im_in, pBuffer, ch_im_in); in arm_convolve_HWC_q7_basic_nonsquare()
125 pBuffer += ch_im_in; in arm_convolve_HWC_q7_basic_nonsquare()
130 if (pBuffer == bufferA + 2 * ch_im_in * dim_kernel_y * dim_kernel_x) in arm_convolve_HWC_q7_basic_nonsquare()
133 …wt, bufferA, ch_im_out, ch_im_in * dim_kernel_y * dim_kernel_x, bias_shift, out_shift, bias, pOut); in arm_convolve_HWC_q7_basic_nonsquare()
156 uint16_t colCnt = ch_im_in * dim_kernel_y * dim_kernel_x >> 2; in arm_convolve_HWC_q7_basic_nonsquare()
173 colCnt = ch_im_in * dim_kernel_y * dim_kernel_x & 0x3; in arm_convolve_HWC_q7_basic_nonsquare()
207 for (l = 0; l < ch_im_in; l++) in arm_convolve_HWC_q7_basic_nonsquare()
209 conv_out += Im_in[(in_row * dim_im_in_x + in_col) * ch_im_in + l] * in arm_convolve_HWC_q7_basic_nonsquare()
[all …]
|
| D | arm_convolve_HWC_q15_basic.c | 76 const uint16_t ch_im_in, in arm_convolve_HWC_q15_basic() argument
116 memset(pBuffer, 0, sizeof(q15_t) * ch_im_in); in arm_convolve_HWC_q15_basic()
123 (q15_t *)Im_in + (i_ker_y * dim_im_in + i_ker_x) * ch_im_in, in arm_convolve_HWC_q15_basic()
124 sizeof(q15_t) * ch_im_in); in arm_convolve_HWC_q15_basic()
126 pBuffer += ch_im_in; 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()
188 for (l = 0; l < ch_im_in; l++) in arm_convolve_HWC_q15_basic()
190 conv_out += Im_in[(in_row * dim_im_in + in_col) * ch_im_in + l] * 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()
|
| D | arm_convolve_HWC_q7_RGB.c | 82 const uint16_t ch_im_in, in arm_convolve_HWC_q7_RGB() argument
109 if (ch_im_in != 3) in arm_convolve_HWC_q7_RGB()
237 if (ch_im_in != 3) in arm_convolve_HWC_q7_RGB()
258 for (l = 0; l < ch_im_in; l++) in arm_convolve_HWC_q7_RGB()
260 conv_out += Im_in[(in_row * dim_im_in + in_col) * ch_im_in + l] * 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()
|
| /hal_nxp-3.5.0/mcux/mcux-sdk/CMSIS/NN/Include/ |
| D | arm_nnfunctions.h | 297 const uint16_t ch_im_in,
338 const uint16_t ch_im_in,
378 const uint16_t ch_im_in,
419 const uint16_t ch_im_in,
467 const uint16_t ch_im_in,
523 const uint16_t ch_im_in,
679 const uint16_t ch_im_in,
721 const uint16_t ch_im_in,
779 const uint16_t ch_im_in,
825 const uint16_t ch_im_in,
[all …]
|
