| /cmsis-nn-latest/Source/PadFunctions/ |
| D | arm_pad_s8.c | 52 int8_t *output, in arm_pad_s8() argument
68 arm_memset_s8(output, pad_value, batch_block_size * pre_pad->n); in arm_pad_s8()
69 output += batch_block_size * pre_pad->n; in arm_pad_s8()
73 arm_memset_s8(output, pad_value, row_block_size * pre_pad->h); in arm_pad_s8()
74 output += row_block_size * pre_pad->h; in arm_pad_s8()
78 arm_memset_s8(output, pad_value, col_block_size * pre_pad->w); in arm_pad_s8()
79 output += col_block_size * pre_pad->w; in arm_pad_s8()
82 arm_memcpy_s8(output, input, input_size->w * input_size->c); in arm_pad_s8()
83 output += input_size->w * input_size->c; in arm_pad_s8()
91 arm_memset_s8(output, pad_value, pre_pad->c); in arm_pad_s8()
[all …]
|
| /cmsis-nn-latest/Source/NNSupportFunctions/ |
| D | arm_nn_lstm_calculate_gate_s16.c | 51 int16_t *output, in arm_nn_lstm_calculate_gate_s16() argument
55 memset(output, 0, params->hidden_size * params->batch_size * sizeof(int16_t)); in arm_nn_lstm_calculate_gate_s16()
60 output, in arm_nn_lstm_calculate_gate_s16()
74 output, in arm_nn_lstm_calculate_gate_s16()
83 …arm_nn_activation_s16(output, output, params->hidden_size * params->batch_size, 0, gate->activatio… in arm_nn_lstm_calculate_gate_s16()
|
| D | arm_nn_lstm_calculate_gate_s8_s16.c | 58 int16_t *output, in arm_nn_lstm_calculate_gate_s8_s16() argument
62 memset(output, 0, params->hidden_size * params->batch_size * sizeof(int16_t)); in arm_nn_lstm_calculate_gate_s8_s16()
67 output, in arm_nn_lstm_calculate_gate_s8_s16()
80 output, in arm_nn_lstm_calculate_gate_s8_s16()
89 …arm_nn_activation_s16(output, output, params->hidden_size * params->batch_size, 0, gate->activatio… in arm_nn_lstm_calculate_gate_s8_s16()
|
| /cmsis-nn-latest/Source/ConcatenationFunctions/ |
| D | arm_concatenation_s8_x.c | 54 int8_t *output, in arm_concatenation_s8_x() argument
60 output += offset_x; in arm_concatenation_s8_x()
67 arm_memcpy_s8(output, input, input_x); in arm_concatenation_s8_x()
69 output += output_x; in arm_concatenation_s8_x()
|
| D | arm_concatenation_s8_z.c | 54 int8_t *output, in arm_concatenation_s8_z() argument
61 output += offset_z * (input_x * input_y); in arm_concatenation_s8_z()
67 arm_memcpy_s8(output, input, input_copy_size); in arm_concatenation_s8_z()
69 output += output_stride; in arm_concatenation_s8_z()
|
| D | arm_concatenation_s8_y.c | 54 int8_t *output, in arm_concatenation_s8_y() argument
62 output += offset_y * input_x; in arm_concatenation_s8_y()
68 arm_memcpy_s8(output, input, input_copy_size); in arm_concatenation_s8_y()
70 output += output_stride; in arm_concatenation_s8_y()
|
| D | arm_concatenation_s8_w.c | 54 int8_t *output, in arm_concatenation_s8_w() argument
59 output += offset_w * (input_x * input_y * input_z); in arm_concatenation_s8_w()
61 arm_memcpy_s8(output, input, input_copy_size); in arm_concatenation_s8_w()
|
| /cmsis-nn-latest/Source/ReshapeFunctions/ |
| D | arm_reshape_s8.c | 50 void arm_reshape_s8(const int8_t *input, int8_t *output, const uint32_t total_size) in arm_reshape_s8() argument
52 arm_memcpy_s8(output, input, total_size); in arm_reshape_s8()
|
| /cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_softmax_s8/ |
| D | test_arm_softmax_s8.c | 35 int8_t output[SOFTMAX_DST_SIZE]; in softmax_arm_softmax_s8() local
39 arm_softmax_s8(input_data, num_rows, row_size, mult, shift, diff_min, output); in softmax_arm_softmax_s8()
40 TEST_ASSERT_TRUE(validate(output, softmax_output_ref, SOFTMAX_DST_SIZE)); in softmax_arm_softmax_s8()
52 int8_t output[SOFTMAX_DST_SIZE]; in softmax_invalid_diff_min_arm_softmax_s8() local
62 arm_softmax_s8(input_data, num_rows, row_size, mult, shift, diff_min, output); in softmax_invalid_diff_min_arm_softmax_s8()
63 TEST_ASSERT_TRUE(validate(output, softmax_expect_invalid_output, SOFTMAX_DST_SIZE)); in softmax_invalid_diff_min_arm_softmax_s8()
|
| /cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_softmax_s8_s16/ |
| D | test_arm_softmax_s8_s16.c | 35 int16_t output[SOFTMAX_S8_S16_DST_SIZE]; in softmax_s8_s16_arm_softmax_s8_s16() local
39 arm_softmax_s8_s16(input_data, num_rows, row_size, mult, shift, diff_min, output); in softmax_s8_s16_arm_softmax_s8_s16()
40 TEST_ASSERT_TRUE(validate_s16(output, softmax_s8_s16_output_ref, SOFTMAX_S8_S16_DST_SIZE)); in softmax_s8_s16_arm_softmax_s8_s16()
52 int16_t output[SOFTMAX_S8_S16_DST_SIZE]; in softmax_s8_s16_invalid_diff_min_arm_softmax_s8_s16() local
62 arm_softmax_s8_s16(input_data, num_rows, row_size, mult, shift, diff_min, output); in softmax_s8_s16_invalid_diff_min_arm_softmax_s8_s16()
63 …TEST_ASSERT_TRUE(validate_s16(output, softmax_s8_s16_expect_invalid_output, SOFTMAX_S8_S16_DST_SIZ… in softmax_s8_s16_invalid_diff_min_arm_softmax_s8_s16()
|
| /cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_softmax_s16/ |
| D | test_arm_softmax_s16.c | 38 int16_t output[SOFTMAX_S16_DST_SIZE]; in softmax_s16_arm_softmax_s16() local
42 arm_softmax_s16(input_data, num_rows, row_size, mult, shift, &softmax_params, output); in softmax_s16_arm_softmax_s16()
43 TEST_ASSERT_TRUE(validate_s16(output, softmax_s16_output_ref, SOFTMAX_S16_DST_SIZE)); in softmax_s16_arm_softmax_s16()
|
| /cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_elementwise_mul_s16/ |
| D | test_arm_elementwise_mul_s16.c | 29 int16_t output[MUL_S16_DST_SIZE] = {0}; in mul_s16_arm_elementwise_mul_s16() local
48 output, in mul_s16_arm_elementwise_mul_s16()
57 TEST_ASSERT_TRUE(validate_s16(output, mul_s16_output_ref, MUL_S16_DST_SIZE)); in mul_s16_arm_elementwise_mul_s16()
63 int16_t output[MUL_S16_SPILL_DST_SIZE] = {0}; in mul_s16_spill_arm_elementwise_mul_s16() local
82 output, in mul_s16_spill_arm_elementwise_mul_s16()
91 TEST_ASSERT_TRUE(validate_s16(output, mul_s16_spill_output_ref, MUL_S16_SPILL_DST_SIZE)); in mul_s16_spill_arm_elementwise_mul_s16()
|
| /cmsis-nn-latest/Source/SoftmaxFunctions/ |
| D | arm_softmax_s8_s16.c | 49 int16_t *output) in arm_softmax_s8_s16() argument
51 … arm_nn_softmax_common_s8(input, num_rows, row_size, mult, shift, diff_min, true, (void *)output); in arm_softmax_s8_s16()
|
| /cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_batch_matmul_s16/ |
| D | test_arm_batch_matmul_s16.c | 64 int16_t output[BATCH_MATMUL_1_S16_DST_SIZE] = {0}; in batch_matmul_1_s16() local
73 …m_params, &quant_params, &lhs_shape_nt, lhs_input, &rhs_shape_t, rhs_input, &output_shape, output); in batch_matmul_1_s16()
76 TEST_ASSERT_TRUE(validate_s16(output, batch_matmul_1_s16_output, output_size)); in batch_matmul_1_s16()
105 int16_t output[BATCH_MATMUL_2_S16_DST_SIZE] = {0}; in batch_matmul_2_s16() local
114 …_params, &quant_params, &lhs_shape_nt, lhs_input, &rhs_shape_nt, rhs_input, &output_shape, output); in batch_matmul_2_s16()
117 TEST_ASSERT_TRUE(validate_s16(output, batch_matmul_2_s16_output, output_size)); in batch_matmul_2_s16()
147 int16_t output[BATCH_MATMUL_3_S16_DST_SIZE] = {0}; in batch_matmul_3_s16() local
156 …mm_params, &quant_params, &lhs_shape_t, lhs_input, &rhs_shape_t, rhs_input, &output_shape, output); in batch_matmul_3_s16()
159 TEST_ASSERT_TRUE(validate_s16(output, batch_matmul_3_s16_output, output_size)); in batch_matmul_3_s16()
189 int16_t output[BATCH_MATMUL_4_S16_DST_SIZE] = {0}; in batch_matmul_4_s16() local
[all …]
|
| /cmsis-nn-latest/Source/BasicMathFunctions/ |
| D | arm_elementwise_mul_acc_s16.c | 53 int16_t *output, in arm_elementwise_mul_acc_s16() argument
84 res_0 = vaddq_s32(res_0, vldrhq_z_s32(output, pred)); in arm_elementwise_mul_acc_s16()
89 vstrhq_p_s32(output, res_0, pred); in arm_elementwise_mul_acc_s16()
93 output += 4; in arm_elementwise_mul_acc_s16()
118 mul_res += output[0]; in arm_elementwise_mul_acc_s16()
132 mul_res += output[1]; in arm_elementwise_mul_acc_s16()
137 arm_nn_write_q15x2_ia(&output, PACK_Q15x2_32x1(mul_1, mul_2)); in arm_elementwise_mul_acc_s16()
152 mul_res += output[0]; in arm_elementwise_mul_acc_s16()
157 *output++ = (int16_t)mul_res; in arm_elementwise_mul_acc_s16()
|
| D | arm_elementwise_mul_s16_s8.c | 50 int8_t *output, in arm_elementwise_mul_s16_s8() argument
66 int8_t *output_ptr = output; in arm_elementwise_mul_s16_s8()
93 output += block_size; in arm_elementwise_mul_s16_s8()
113 arm_nn_write_s8x2_ia(&output, mul); in arm_elementwise_mul_s16_s8()
117 for (int j = 0; j < loop_count; j++, input_1_vect++, input_2_vect++, output++) in arm_elementwise_mul_s16_s8()
125 *output = (int8_t)mul_res; in arm_elementwise_mul_s16_s8()
130 output += (batch_offset - 1) * block_size; in arm_elementwise_mul_s16_s8()
|
| /cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_depthwise_conv_s4_opt/ |
| D | test_arm_depthwise_conv_s4_opt.c | 33 int8_t output[DEPTHWISE_INT4_1_DST_SIZE] = {0}; in depthwise_int4_1_arm_depthwise_conv_s4_opt() local
89 output); in depthwise_int4_1_arm_depthwise_conv_s4_opt()
98 TEST_ASSERT_TRUE(validate(output, depthwise_int4_1_output_ref, DEPTHWISE_INT4_1_DST_SIZE)); in depthwise_int4_1_arm_depthwise_conv_s4_opt()
111 output); in depthwise_int4_1_arm_depthwise_conv_s4_opt()
119 TEST_ASSERT_TRUE(validate(output, depthwise_int4_1_output_ref, DEPTHWISE_INT4_1_DST_SIZE)); in depthwise_int4_1_arm_depthwise_conv_s4_opt()
120 memset(output, 0, DEPTHWISE_INT4_1_DST_SIZE); in depthwise_int4_1_arm_depthwise_conv_s4_opt()
134 output); in depthwise_int4_1_arm_depthwise_conv_s4_opt()
143 TEST_ASSERT_TRUE(validate(output, depthwise_int4_1_output_ref, DEPTHWISE_INT4_1_DST_SIZE)); in depthwise_int4_1_arm_depthwise_conv_s4_opt()
144 memset(output, 0, DEPTHWISE_INT4_1_DST_SIZE); in depthwise_int4_1_arm_depthwise_conv_s4_opt()
150 int8_t output[DEPTHWISE_INT4_2_DST_SIZE] = {}; in depthwise_int4_2_arm_depthwise_conv_s4_opt() local
[all …]
|
| /cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_depthwise_conv_s4/ |
| D | test_arm_depthwise_conv_s4.c | 35 int8_t output[DEPTHWISE_INT4_GENERIC_DST_SIZE] = {0}; in depthwise_int4_generic_arm_depthwise_conv_s4() local
90 output); in depthwise_int4_generic_arm_depthwise_conv_s4()
99 …TEST_ASSERT_TRUE(validate(output, depthwise_int4_generic_output_ref, DEPTHWISE_INT4_GENERIC_DST_SI… in depthwise_int4_generic_arm_depthwise_conv_s4()
100 memset(output, 0, DEPTHWISE_INT4_GENERIC_DST_SIZE); in depthwise_int4_generic_arm_depthwise_conv_s4()
113 output); in depthwise_int4_generic_arm_depthwise_conv_s4()
121 …TEST_ASSERT_TRUE(validate(output, depthwise_int4_generic_output_ref, DEPTHWISE_INT4_GENERIC_DST_SI… in depthwise_int4_generic_arm_depthwise_conv_s4()
122 memset(output, 0, DEPTHWISE_INT4_GENERIC_DST_SIZE); in depthwise_int4_generic_arm_depthwise_conv_s4()
128 int8_t output[DEPTHWISE_INT4_GENERIC_2_DST_SIZE] = {0}; in depthwise_int4_generic_2_arm_depthwise_conv_s4() local
183 output); in depthwise_int4_generic_2_arm_depthwise_conv_s4()
192 …TEST_ASSERT_TRUE(validate(output, depthwise_int4_generic_2_output_ref, DEPTHWISE_INT4_GENERIC_2_DS… in depthwise_int4_generic_2_arm_depthwise_conv_s4()
[all …]
|
| /cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_elementwise_mul_s8/ |
| D | test_arm_elementwise_mul_s8.c | 28 int8_t output[MUL_DST_SIZE] = {0}; in mul_arm_elementwise_mul_s8() local
47 output, in mul_arm_elementwise_mul_s8()
56 TEST_ASSERT_TRUE(validate(output, mul_output_ref, MUL_DST_SIZE)); in mul_arm_elementwise_mul_s8()
|
| /cmsis-nn-latest/Source/ConvolutionFunctions/ |
| D | arm_depthwise_conv_wrapper_s16.c | 60 int16_t *output) in arm_depthwise_conv_wrapper_s16() argument
76 output); in arm_depthwise_conv_wrapper_s16()
90 output); in arm_depthwise_conv_wrapper_s16()
|
| /cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_convolve_s16/ |
| D | test_arm_convolve_s16.c | 42 int16_t output[INT16XINT8_DST_SIZE] = {0}; in int16xint8_arm_convolve_s16() local
96 output); in int16xint8_arm_convolve_s16()
104 TEST_ASSERT_TRUE(validate_s16(output, output_ref, output_ref_size)); in int16xint8_arm_convolve_s16()
105 memset(output, 0, sizeof(output)); in int16xint8_arm_convolve_s16()
120 output); in int16xint8_arm_convolve_s16()
127 TEST_ASSERT_TRUE(validate_s16(output, output_ref, output_ref_size)); in int16xint8_arm_convolve_s16()
132 int16_t output[REQUANTIZE_S64_DST_SIZE] = {0}; in requantize_s64_arm_convolve_s16() local
186 output); in requantize_s64_arm_convolve_s16()
193 TEST_ASSERT_TRUE(validate_s16(output, output_ref, output_ref_size)); in requantize_s64_arm_convolve_s16()
194 memset(output, 0, sizeof(output)); in requantize_s64_arm_convolve_s16()
[all …]
|
| /cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_batch_matmul_s8/ |
| D | test_arm_batch_matmul_s8.c | 64 int8_t output[BATCH_MATMUL_1_S8_DST_SIZE] = {0}; in batch_matmul_1_s8() local
74 …m_params, &quant_params, &lhs_shape_nt, lhs_input, &rhs_shape_t, rhs_input, &output_shape, output); in batch_matmul_1_s8()
83 TEST_ASSERT_TRUE(validate(output, batch_matmul_1_s8_output, output_size)); in batch_matmul_1_s8()
112 int8_t output[BATCH_MATMUL_2_S8_DST_SIZE] = {0}; in batch_matmul_2_s8() local
122 …_params, &quant_params, &lhs_shape_nt, lhs_input, &rhs_shape_nt, rhs_input, &output_shape, output); in batch_matmul_2_s8()
131 TEST_ASSERT_TRUE(validate(output, batch_matmul_2_s8_output, output_size)); in batch_matmul_2_s8()
161 int8_t output[BATCH_MATMUL_3_S8_DST_SIZE] = {0}; in batch_matmul_3_s8() local
171 …mm_params, &quant_params, &lhs_shape_t, lhs_input, &rhs_shape_t, rhs_input, &output_shape, output); in batch_matmul_3_s8()
180 TEST_ASSERT_TRUE(validate(output, batch_matmul_3_s8_output, output_size)); in batch_matmul_3_s8()
210 int8_t output[BATCH_MATMUL_4_S8_DST_SIZE] = {0}; in batch_matmul_4_s8() local
[all …]
|
| /cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_elementwise_add_s16/ |
| D | test_arm_elementwise_add_s16.c | 29 int16_t output[ADD_S16_DST_SIZE] = {0}; in add_s16_arm_elementwise_add_s16() local
59 output, in add_s16_arm_elementwise_add_s16()
68 TEST_ASSERT_TRUE(validate_s16(output, add_s16_output_ref, ADD_S16_DST_SIZE)); in add_s16_arm_elementwise_add_s16()
74 int16_t output[ADD_S16_SPILL_DST_SIZE] = {0}; in add_s16_spill_arm_elementwise_add_s16() local
104 output, in add_s16_spill_arm_elementwise_add_s16()
113 TEST_ASSERT_TRUE(validate_s16(output, add_s16_spill_output_ref, ADD_S16_SPILL_DST_SIZE)); in add_s16_spill_arm_elementwise_add_s16()
|
| /cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_maximum_minimum_s8/ |
| D | test_arm_maximum_s8.c | 38 int8_t output[MAXIMUM_SCALAR_1_INT8_DST_SIZE] = {0}; in maximum_scalar_1_int8() local
68 …arm_maximum_s8(&ctx, input_1_data, &input_1_dims, input_2_data, &input_2_dims, output, &output_dim… in maximum_scalar_1_int8()
79 TEST_ASSERT_TRUE(validate(output, output_ref_data, dst_size)); in maximum_scalar_1_int8()
86 int8_t output[MAXIMUM_SCALAR_2_INT8_DST_SIZE] = {0}; in maximum_scalar_2_int8() local
116 …arm_maximum_s8(&ctx, input_1_data, &input_1_dims, input_2_data, &input_2_dims, output, &output_dim… in maximum_scalar_2_int8()
127 TEST_ASSERT_TRUE(validate(output, output_ref_data, dst_size)); in maximum_scalar_2_int8()
133 int8_t output[MAXIMUM_NO_BROADCAST_INT8_DST_SIZE] = {0}; in maximum_no_broadcast_int8() local
163 …arm_maximum_s8(&ctx, input_1_data, &input_1_dims, input_2_data, &input_2_dims, output, &output_dim… in maximum_no_broadcast_int8()
174 TEST_ASSERT_TRUE(validate(output, output_ref_data, dst_size)); in maximum_no_broadcast_int8()
180 int8_t output[MAXIMUM_BROADCAST_BATCH_INT8_DST_SIZE] = {0}; in maximum_broadcast_batch_int8() local
[all …]
|
| D | test_arm_minimum_s8.c | 38 int8_t output[MINIMUM_SCALAR_1_INT8_DST_SIZE] = {0}; in minimum_scalar_1_int8() local
68 …arm_minimum_s8(&ctx, input_1_data, &input_1_dims, input_2_data, &input_2_dims, output, &output_dim… in minimum_scalar_1_int8()
79 TEST_ASSERT_TRUE(validate(output, output_ref_data, dst_size)); in minimum_scalar_1_int8()
86 int8_t output[MINIMUM_SCALAR_2_INT8_DST_SIZE] = {0}; in minimum_scalar_2_int8() local
116 …arm_minimum_s8(&ctx, input_1_data, &input_1_dims, input_2_data, &input_2_dims, output, &output_dim… in minimum_scalar_2_int8()
127 TEST_ASSERT_TRUE(validate(output, output_ref_data, dst_size)); in minimum_scalar_2_int8()
133 int8_t output[MINIMUM_NO_BROADCAST_INT8_DST_SIZE] = {0}; in minimum_no_broadcast_int8() local
163 …arm_minimum_s8(&ctx, input_1_data, &input_1_dims, input_2_data, &input_2_dims, output, &output_dim… in minimum_no_broadcast_int8()
174 TEST_ASSERT_TRUE(validate(output, output_ref_data, dst_size)); in minimum_no_broadcast_int8()
180 int8_t output[MINIMUM_BROADCAST_BATCH_INT8_DST_SIZE] = {0}; in minimum_broadcast_batch_int8() local
[all …]
|
