| /hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Source/ConvolutionFunctions/ |
| D | arm_convolve_s8.c | 74 const uint16_t output_ch = output_dims->c; in arm_convolve_s8() local
129 for (int i_out_ch = 0; i_out_ch < output_ch; i_out_ch++) in arm_convolve_s8()
151 … const uint32x4_t scatter_offset = {0, output_ch, output_ch * 2, output_ch * 3}; in arm_convolve_s8()
155 out += (3 * output_ch); in arm_convolve_s8()
163 output_ch, in arm_convolve_s8()
186 output_ch, in arm_convolve_s8()
239 output_ch, in arm_convolve_s8()
261 for (i = 0; i < output_ch; i++) in arm_convolve_s8()
313 for (i_out_ch = 0; i_out_ch < output_ch; i_out_ch++) in arm_convolve_s8()
353 … output_data[i_out_ch + (i_out_y * output_x + i_out_x) * output_ch] = (int8_t)conv_out; in arm_convolve_s8()
[all …]
|
| D | arm_nn_mat_mult_s8.c | 41 const uint16_t output_ch, in arm_nn_mat_mult_s8() argument
58 for (int i_out_ch = 0; i_out_ch < output_ch; i_out_ch++) in arm_nn_mat_mult_s8()
115 const uint32x4_t scatter_offset = {0, output_ch, output_ch * 2, output_ch * 3}; in arm_nn_mat_mult_s8()
118 out += 4 * output_ch; in arm_nn_mat_mult_s8()
124 for (int i_out_ch = 0; i_out_ch < output_ch; i_out_ch++) in arm_nn_mat_mult_s8()
158 out += output_ch; in arm_nn_mat_mult_s8()
166 (void)output_ch; in arm_nn_mat_mult_s8()
|
| D | arm_convolve_1x1_s8_fast.c | 78 const int32_t output_ch = output_dims->c; in arm_convolve_1x1_s8_fast() local
89 for (int i_out_ch = 0; i_out_ch < output_ch; i_out_ch++) in arm_convolve_1x1_s8_fast()
114 0, (uint32_t)output_ch, (uint32_t)output_ch * 2, (uint32_t)output_ch * 3}; in arm_convolve_1x1_s8_fast()
118 output_data += (3 * output_ch); in arm_convolve_1x1_s8_fast()
124 for (int i_out_ch = 0; i_out_ch < output_ch; i_out_ch++) in arm_convolve_1x1_s8_fast()
|
| D | arm_nn_mat_mult_kernel_s8_s16.c | 42 const uint16_t output_ch, in arm_nn_mat_mult_kernel_s8_s16() argument
57 q7_t *out_1 = out_0 + output_ch; in arm_nn_mat_mult_kernel_s8_s16()
61 int32_t row_count = output_ch / ROW_PER_LOOP; in arm_nn_mat_mult_kernel_s8_s16()
171 row_count = output_ch & (ROW_PER_LOOP - 1); in arm_nn_mat_mult_kernel_s8_s16()
175 ip_a0_s8 = input_a + num_col_a * (output_ch & ~3); in arm_nn_mat_mult_kernel_s8_s16()
221 q7_t *out_1 = out_0 + output_ch; in arm_nn_mat_mult_kernel_s8_s16()
224 uint16_t row_count = output_ch / 2; in arm_nn_mat_mult_kernel_s8_s16()
317 if (output_ch & 0x1) in arm_nn_mat_mult_kernel_s8_s16()
372 out_0 += output_ch; in arm_nn_mat_mult_kernel_s8_s16()
379 (void)output_ch; in arm_nn_mat_mult_kernel_s8_s16()
|
| D | arm_convolve_1_x_n_s8.c | 76 const uint16_t output_ch = output_dims->c; in arm_convolve_1_x_n_s8() local
102 for (int i_out_ch = 0; i_out_ch < output_ch; i_out_ch++) in arm_convolve_1_x_n_s8()
164 const uint32x4_t scatter_offset = {0, output_ch, output_ch * 2, output_ch * 3}; in arm_convolve_1_x_n_s8()
168 output_data += (3 * output_ch); in arm_convolve_1_x_n_s8()
|
| D | arm_nn_mat_mult_kernel_s8_s16_reordered.c | 45 const uint16_t output_ch, in arm_nn_mat_mult_kernel_s8_s16_reordered() argument
57 q7_t *out_1 = out_0 + output_ch; in arm_nn_mat_mult_kernel_s8_s16_reordered()
60 uint16_t row_count = output_ch / 2; in arm_nn_mat_mult_kernel_s8_s16_reordered()
138 if (output_ch & 1) in arm_nn_mat_mult_kernel_s8_s16_reordered()
182 out_0 += output_ch; in arm_nn_mat_mult_kernel_s8_s16_reordered()
189 (void)output_ch; in arm_nn_mat_mult_kernel_s8_s16_reordered()
|
| D | arm_depthwise_conv_u8_basic_ver1.c | 48 const int32_t output_ch, in depthwise_conv_u8_mult_4() argument
72 for (int32_t in_ch = 0, out_ch = 0, ker_w_start = MAX(0, -in_w); out_ch < output_ch; in depthwise_conv_u8_mult_4()
86 … int32_t ker_idx = ker_h * (output_ch * kernel_x) + ker_w_start * output_ch + out_ch; in depthwise_conv_u8_mult_4()
90 ++ker_w, ker_idx += output_ch) in depthwise_conv_u8_mult_4()
137 const int32_t output_ch, in depthwise_conv_u8_generic() argument
157 (void)output_ch; in depthwise_conv_u8_generic()
|
| D | arm_depthwise_conv_s8.c | 48 const int32_t output_ch, in depthwise_conv_s8_mult_4() argument
71 for (int32_t in_ch = 0, out_ch = 0, ker_w_start = MAX(0, -in_w); out_ch < output_ch; in depthwise_conv_s8_mult_4()
85 … int32_t ker_idx = ker_h * (output_ch * kernel_x) + ker_w_start * output_ch + out_ch; in depthwise_conv_s8_mult_4()
89 ++ker_w, ker_idx += output_ch) in depthwise_conv_s8_mult_4()
148 const uint16_t output_ch, in depthwise_conv_s8_generic() argument
167 (void)output_ch; in depthwise_conv_s8_generic()
|
| D | arm_depthwise_conv_s8_opt.c | 65 const int32_t output_ch = output_dims->c; in arm_depthwise_conv_s8_opt() local
68 if (input_ch != output_ch) in arm_depthwise_conv_s8_opt()
267 row_count = output_ch / 4; in arm_depthwise_conv_s8_opt()
366 row_count = output_ch & 0x3; in arm_depthwise_conv_s8_opt()
|
| D | arm_depthwise_conv_3x3_s8.c | 70 const int32_t output_ch = output_dims->c; in arm_depthwise_conv_3x3_s8() local
85 if (input_ch != output_ch) in arm_depthwise_conv_3x3_s8()
|
| /hal_nxp-latest/mcux/mcux-sdk/CMSIS/NN/Include/ |
| D | arm_nnfunctions.h | 1354 const uint16_t output_ch,
1379 const uint16_t output_ch,
|
| D | arm_nnsupportfunctions.h | 227 const uint16_t output_ch,
|
