Home
last modified time | relevance | path

Searched refs:ch_im_out (Results 1 – 14 of 14) sorted by relevance

/hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Source/ConvolutionFunctions/
Darm_convolve_HWC_q7_fast.c95 const uint16_t ch_im_out, 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()
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()
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()
219 … wt, bufferA, ch_im_out, ch_im_in * dim_kernel * dim_kernel, bias_shift, out_shift, bias, pOut); in arm_convolve_HWC_q7_fast()
250 … wt, bufferA, ch_im_out, ch_im_in * dim_kernel * dim_kernel, bias_shift, out_shift, bias, pOut); in arm_convolve_HWC_q7_fast()
283 … wt, bufferA, ch_im_out, ch_im_in * dim_kernel * dim_kernel, bias_shift, out_shift, bias, pOut); in arm_convolve_HWC_q7_fast()
296 for (i = 0; i < ch_im_out; i++) in arm_convolve_HWC_q7_fast()
337 if (ch_im_in % 4 != 0 || ch_im_out % 2 != 0) in arm_convolve_HWC_q7_fast()
343 for (i = 0; i < ch_im_out; i++) in arm_convolve_HWC_q7_fast()
[all …]
Darm_convolve_HWC_q7_fast_nonsquare.c79 const uint16_t ch_im_out, 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()
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()
189 …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()
212 …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()
245 …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()
280 …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()
292 for (i = 0; i < ch_im_out; i++) in arm_convolve_HWC_q7_fast_nonsquare()
334 if (ch_im_in % 4 != 0 || ch_im_out % 2 != 0) in arm_convolve_HWC_q7_fast_nonsquare()
340 for (i = 0; i < ch_im_out; i++) in arm_convolve_HWC_q7_fast_nonsquare()
[all …]
Darm_convolve_HWC_q15_fast.c83 const uint16_t ch_im_out, 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()
144 q15_t *pOut2 = pOut + ch_im_out; in arm_convolve_HWC_q15_fast()
147 for (i = 0; i < ch_im_out; i += 2) in arm_convolve_HWC_q15_fast()
201 pOut += ch_im_out; in arm_convolve_HWC_q15_fast()
215 if (ch_im_in % 2 != 0 || ch_im_out % 2 != 0) in arm_convolve_HWC_q15_fast()
221 for (i = 0; i < ch_im_out; i++) in arm_convolve_HWC_q15_fast()
244 … Im_out[i + (j * dim_im_out + k) * ch_im_out] = (q15_t)__SSAT((conv_out >> out_shift), 16); in arm_convolve_HWC_q15_fast()
Darm_convolve_HWC_q15_fast_nonsquare.c89 const uint16_t ch_im_out, 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()
156 q15_t *pOut2 = pOut + ch_im_out; in arm_convolve_HWC_q15_fast_nonsquare()
159 for (i = 0; i < ch_im_out; i += 2) in arm_convolve_HWC_q15_fast_nonsquare()
213 pOut += ch_im_out; in arm_convolve_HWC_q15_fast_nonsquare()
227 if (ch_im_in % 2 != 0 || ch_im_out % 2 != 0) in arm_convolve_HWC_q15_fast_nonsquare()
233 for (i = 0; i < ch_im_out; i++) in arm_convolve_HWC_q15_fast_nonsquare()
257 … Im_out[i + (j * dim_im_out_x + k) * ch_im_out] = (q15_t)__SSAT((conv_out >> out_shift), 16); in arm_convolve_HWC_q15_fast_nonsquare()
Darm_convolve_1x1_HWC_q7_fast_nonsquare.c86 const uint16_t ch_im_out, 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()
136 wt, bufferA, ch_im_out, ch_im_in, bias_shift, out_shift, bias, pOut); in arm_convolve_1x1_HWC_q7_fast_nonsquare()
147 for (i_ch_out = 0; i_ch_out < ch_im_out; i_ch_out++) 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()
197 for (i = 0; i < ch_im_out; i++) in arm_convolve_1x1_HWC_q7_fast_nonsquare()
222 … Im_out[i + (j * dim_im_out_x + k) * ch_im_out] = (q7_t)__SSAT((conv_out >> out_shift), 8); in arm_convolve_1x1_HWC_q7_fast_nonsquare()
Darm_depthwise_separable_conv_HWC_q7.c86 const uint16_t ch_im_out, in arm_depthwise_separable_conv_HWC_q7() argument
112 if (ch_im_in != ch_im_out) in arm_depthwise_separable_conv_HWC_q7()
142 rowCnt = ch_im_out >> 2; in arm_depthwise_separable_conv_HWC_q7()
346 rowCnt = ch_im_out & 0x3; in arm_depthwise_separable_conv_HWC_q7()
382 if (ch_im_in != ch_im_out) in arm_depthwise_separable_conv_HWC_q7()
391 for (i_ch_out = 0; i_ch_out < ch_im_out; i_ch_out++) 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()
408 Im_out[(i_out_y * dim_im_out + i_out_x) * ch_im_out + i_ch_out] = in arm_depthwise_separable_conv_HWC_q7()
Darm_depthwise_separable_conv_HWC_q7_nonsquare.c79 const uint16_t ch_im_out, 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()
152 rowCnt = ch_im_out >> 2; in arm_depthwise_separable_conv_HWC_q7_nonsquare()
350 rowCnt = ch_im_out & 0x3; in arm_depthwise_separable_conv_HWC_q7_nonsquare()
387 if (ch_im_in != ch_im_out) in arm_depthwise_separable_conv_HWC_q7_nonsquare()
396 for (i_ch_out = 0; i_ch_out < ch_im_out; i_ch_out++) in arm_depthwise_separable_conv_HWC_q7_nonsquare()
409 wt[(i_ker_y * dim_kernel_x + i_ker_x) * ch_im_out + i_ch_out]; in arm_depthwise_separable_conv_HWC_q7_nonsquare()
413 Im_out[(i_out_y * dim_im_out_x + i_out_x) * ch_im_out + i_ch_out] = in arm_depthwise_separable_conv_HWC_q7_nonsquare()
Darm_nn_mat_mult_kernel_q7_q15.c42 const uint16_t ch_im_out, in arm_nn_mat_mult_kernel_q7_q15() argument
51 q7_t *pOut2 = pOut + ch_im_out; in arm_nn_mat_mult_kernel_q7_q15()
54 uint16_t rowCnt = ch_im_out >> 1; in arm_nn_mat_mult_kernel_q7_q15()
123 if (ch_im_out & 0x1) in arm_nn_mat_mult_kernel_q7_q15()
170 pOut += ch_im_out; in arm_nn_mat_mult_kernel_q7_q15()
177 (void)ch_im_out; in arm_nn_mat_mult_kernel_q7_q15()
Darm_nn_mat_mult_kernel_q7_q15_reordered.c42 const uint16_t ch_im_out, in arm_nn_mat_mult_kernel_q7_q15_reordered() argument
52 q7_t *pOut2 = pOut + ch_im_out; in arm_nn_mat_mult_kernel_q7_q15_reordered()
56 for (i = 0; i < ch_im_out; i += 2) in arm_nn_mat_mult_kernel_q7_q15_reordered()
121 pOut += ch_im_out; in arm_nn_mat_mult_kernel_q7_q15_reordered()
128 (void)ch_im_out; in arm_nn_mat_mult_kernel_q7_q15_reordered()
Darm_convolve_HWC_q7_basic.c78 const uint16_t ch_im_out, in arm_convolve_HWC_q7_basic() argument
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()
146 for (i = 0; i < ch_im_out; i++) in arm_convolve_HWC_q7_basic()
190 for (i = 0; i < ch_im_out; i++) in arm_convolve_HWC_q7_basic()
214 … Im_out[i + (j * dim_im_out + k) * ch_im_out] = (q7_t)__SSAT((conv_out >> out_shift), 8); in arm_convolve_HWC_q7_basic()
Darm_convolve_HWC_q7_basic_nonsquare.c73 const uint16_t ch_im_out, in arm_convolve_HWC_q7_basic_nonsquare() argument
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()
147 for (i = 0; i < ch_im_out; i++) in arm_convolve_HWC_q7_basic_nonsquare()
191 for (i = 0; i < ch_im_out; i++) in arm_convolve_HWC_q7_basic_nonsquare()
216 … Im_out[i + (j * dim_im_out_x + k) * ch_im_out] = (q7_t)__SSAT((conv_out >> out_shift), 8); in arm_convolve_HWC_q7_basic_nonsquare()
Darm_convolve_HWC_q7_RGB.c84 const uint16_t ch_im_out, in arm_convolve_HWC_q7_RGB() argument
182 … wt, bufferA, ch_im_out, 3 * dim_kernel * dim_kernel, bias_shift, out_shift, bias, pOut); in arm_convolve_HWC_q7_RGB()
196 for (i = 0; i < ch_im_out; i++) in arm_convolve_HWC_q7_RGB()
242 for (i = 0; i < ch_im_out; i++) in arm_convolve_HWC_q7_RGB()
266 … Im_out[i + (j * dim_im_out + k) * ch_im_out] = (q7_t)__SSAT((conv_out >> out_shift), 8); in arm_convolve_HWC_q7_RGB()
Darm_convolve_HWC_q15_basic.c78 const uint16_t ch_im_out, in arm_convolve_HWC_q15_basic() argument
131 for (i = 0; i < ch_im_out; i++) in arm_convolve_HWC_q15_basic()
173 for (i = 0; i < ch_im_out; i++) in arm_convolve_HWC_q15_basic()
196 … Im_out[i + (j * dim_im_out + k) * ch_im_out] = (q15_t)__SSAT((conv_out >> out_shift), 16); in arm_convolve_HWC_q15_basic()
/hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Include/
Darm_nnfunctions.h299 const uint16_t ch_im_out,
340 const uint16_t ch_im_out,
380 const uint16_t ch_im_out,
421 const uint16_t ch_im_out,
469 const uint16_t ch_im_out,
525 const uint16_t ch_im_out,
681 const uint16_t ch_im_out,
723 const uint16_t ch_im_out,
781 const uint16_t ch_im_out,
827 const uint16_t ch_im_out,
[all …]