| /cmsis-3.4.0/CMSIS/NN/Source/NNSupportFunctions/ |
| D | arm_nn_mat_mul_core_4x_s8.c | 52 int32_t *const output) in arm_nn_mat_mul_core_4x_s8() argument
104 output[0] = acc_n0; in arm_nn_mat_mul_core_4x_s8()
105 output[1] = acc_n1; in arm_nn_mat_mul_core_4x_s8()
106 output[2] = acc_n2; in arm_nn_mat_mul_core_4x_s8()
107 output[3] = acc_n3; in arm_nn_mat_mul_core_4x_s8()
|
| D | arm_nn_mat_mul_core_1x_s8.c | 52 int32_t *const output) in arm_nn_mat_mul_core_1x_s8() argument
80 *output = acc_n0; in arm_nn_mat_mul_core_1x_s8()
|
| /cmsis-3.4.0/CMSIS/NN/Source/ConcatenationFunctions/ |
| D | arm_concatenation_s8_x.c | 53 int8_t *output, in arm_concatenation_s8_x() argument
59 output += offset_x; in arm_concatenation_s8_x()
66 memcpy(output, input, input_x); in arm_concatenation_s8_x()
68 output += output_x; in arm_concatenation_s8_x()
|
| D | arm_concatenation_s8_z.c | 53 int8_t *output, in arm_concatenation_s8_z() argument
60 output += offset_z * (input_x * input_y); in arm_concatenation_s8_z()
66 memcpy(output, input, input_copy_size); in arm_concatenation_s8_z()
68 output += output_stride; in arm_concatenation_s8_z()
|
| D | arm_concatenation_s8_y.c | 53 int8_t *output, in arm_concatenation_s8_y() argument
61 output += offset_y * input_x; in arm_concatenation_s8_y()
67 memcpy(output, input, input_copy_size); in arm_concatenation_s8_y()
69 output += output_stride; in arm_concatenation_s8_y()
|
| D | arm_concatenation_s8_w.c | 53 int8_t *output, in arm_concatenation_s8_w() argument
58 output += offset_w * (input_x * input_y * input_z); in arm_concatenation_s8_w()
60 memcpy(output, input, input_copy_size); in arm_concatenation_s8_w()
|
| /cmsis-3.4.0/CMSIS/NN/Source/ConvolutionFunctions/ |
| D | arm_depthwise_conv_s8.c | 57 int8_t *output, in depthwise_conv_s8_mult_4() argument
108 vstrbq_s32(output, res); in depthwise_conv_s8_mult_4()
109 output += 4; in depthwise_conv_s8_mult_4()
130 output[out_idx++] = (int8_t)out_buff[0]; in depthwise_conv_s8_mult_4()
131 output[out_idx++] = (int8_t)out_buff[1]; in depthwise_conv_s8_mult_4()
132 output[out_idx++] = (int8_t)out_buff[2]; in depthwise_conv_s8_mult_4()
133 output[out_idx++] = (int8_t)out_buff[3]; in depthwise_conv_s8_mult_4()
157 q7_t *output, in depthwise_conv_s8_generic() argument
212 output[i_out++] = acc_0; in depthwise_conv_s8_generic()
239 q7_t *output) in arm_depthwise_conv_s8() argument
[all …]
|
| D | arm_depthwise_conv_wrapper_s8.c | 59 q7_t *output) in arm_depthwise_conv_wrapper_s8() argument
77 output); in arm_depthwise_conv_wrapper_s8()
92 output); in arm_depthwise_conv_wrapper_s8()
107 output); in arm_depthwise_conv_wrapper_s8()
|
| D | arm_depthwise_conv_3x3_s8.c | 62 q7_t *output) in arm_depthwise_conv_3x3_s8() argument
166 output[out_idx++] = (int8_t)out_buff0; in arm_depthwise_conv_3x3_s8()
167 output[out_idx++] = (int8_t)out_buff1; in arm_depthwise_conv_3x3_s8()
168 output[out_idx++] = (int8_t)out_buff2; in arm_depthwise_conv_3x3_s8()
169 output[out_idx++] = (int8_t)out_buff3; in arm_depthwise_conv_3x3_s8()
201 output[out_idx++] = (int8_t)out_buff; in arm_depthwise_conv_3x3_s8()
|
| D | arm_depthwise_conv_s8_opt.c | 61 q7_t *output) in arm_depthwise_conv_s8_opt() argument
95 q7_t *out = output; in arm_depthwise_conv_s8_opt()
344 *output++ = (q7_t)sum; in arm_depthwise_conv_s8_opt()
350 *output++ = (q7_t)sum_2; in arm_depthwise_conv_s8_opt()
355 *output++ = (q7_t)sum_3; in arm_depthwise_conv_s8_opt()
361 *output++ = (q7_t)sum_4; in arm_depthwise_conv_s8_opt()
383 *output++ = (q7_t)sum; in arm_depthwise_conv_s8_opt()
405 output); in arm_depthwise_conv_s8_opt()
|
| /cmsis-3.4.0/CMSIS/NN/Source/SoftmaxFunctions/ |
| D | arm_softmax_u8.c | 50 uint8_t *output) in arm_softmax_u8() argument
90 output[col] = (uint8_t)CLAMP(res, (int32_t)255, (int32_t)0); in arm_softmax_u8()
94 output[col] = 0; in arm_softmax_u8()
98 output += row_size; in arm_softmax_u8()
|
| D | arm_softmax_s8.c | 92 int8_t *output) in arm_softmax_s8() argument
184 vstrbq_s32(&output[idx], tmp_res); in arm_softmax_s8()
196 output[tail_idx + i] = (int8_t)CLAMP(res, (int32_t)ACT_MAX, (int32_t)ACT_MIN); in arm_softmax_s8()
200 output[tail_idx + i] = ACT_MIN; in arm_softmax_s8()
205 output += row_size; in arm_softmax_s8()
246 output[col] = (int8_t)CLAMP(res, (int32_t)127, (int32_t)-128); in arm_softmax_s8()
250 output[col] = -128; in arm_softmax_s8()
254 output += row_size; in arm_softmax_s8()
|
| /cmsis-3.4.0/CMSIS/NN/Source/FullyConnectedFunctions/ |
| D | arm_fully_connected_s8.c | 60 q7_t *output) in arm_fully_connected_s8() argument
73 output, in arm_fully_connected_s8()
84 output += output_dims->c; in arm_fully_connected_s8()
|
| /cmsis-3.4.0/CMSIS/NN/Source/BasicMathFunctions/ |
| D | arm_elementwise_mul_s8.c | 54 int8_t *output, in arm_elementwise_mul_s8() argument
88 vstrbq_p_s32(output, res_0, p); in arm_elementwise_mul_s8()
91 output += 4; in arm_elementwise_mul_s8()
166 write_q7x4_ia(&output, __PACKq7(r1, r2, r3, r4)); in arm_elementwise_mul_s8()
189 *output++ = (q7_t)mul_res; in arm_elementwise_mul_s8()
|
| D | arm_elementwise_add_s8.c | 79 int8_t *output, in arm_elementwise_add_s8() argument
119 vstrbq_p_s32(output, vect_1, p); in arm_elementwise_add_s8()
121 output += 4; in arm_elementwise_add_s8()
212 write_q7x4_ia(&output, __PACKq7(r1, r2, r3, r4)); in arm_elementwise_add_s8()
242 *output++ = (q7_t)sum; in arm_elementwise_add_s8()
|
| /cmsis-3.4.0/CMSIS/NN/Source/ReshapeFunctions/ |
| D | arm_reshape_s8.c | 49 void arm_reshape_s8(const int8_t *input, int8_t *output, const uint32_t total_size) in arm_reshape_s8() argument
51 memcpy(output, input, total_size); in arm_reshape_s8()
|
| /cmsis-3.4.0/CMSIS/NN/Source/ActivationFunctions/ |
| D | arm_relu_q7.c | 62 q7_t *output = data; in arm_relu_q7() local
77 write_q7x4_ia(&output, in & (~mask)); in arm_relu_q7()
|
| /cmsis-3.4.0/CMSIS/NN/Include/ |
| D | arm_nnfunctions.h | 1462 int8_t *output,
1491 int8_t *output,
1760 int8_t *output);
1783 uint8_t *output);
1837 uint8_t *output,
1859 void arm_reshape_s8(const int8_t *input, int8_t *output, const uint32_t total_size);
1907 int8_t *output,
1952 int8_t *output,
1997 int8_t *output,
2038 int8_t *output,
|
| D | arm_nnsupportfunctions.h | 110 void arm_nn_add_q7(const q7_t *input, q31_t *output, uint32_t block_size);
261 int32_t *const output);
290 int32_t *const output);
|
