Home
last modified time | relevance | path

Searched refs:output (Results 1 – 25 of 81) sorted by relevance

1234

/cmsis-nn-latest/Source/NNSupportFunctions/
Darm_nn_lstm_calculate_gate_s16.c51 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()
Darm_nn_lstm_calculate_gate_s8_s16.c58 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/
Darm_concatenation_s8_x.c54 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()
Darm_concatenation_s8_z.c54 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()
Darm_concatenation_s8_y.c54 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()
Darm_concatenation_s8_w.c54 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/Tests/UnitTest/TestCases/test_arm_softmax_s8/
Dtest_arm_softmax_s8.c35 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/
Dtest_arm_softmax_s8_s16.c35 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/Source/ReshapeFunctions/
Darm_reshape_s8.c50 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_elementwise_mul_s16/
Dtest_arm_elementwise_mul_s16.c29 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/Tests/UnitTest/TestCases/test_arm_softmax_s16/
Dtest_arm_softmax_s16.c38 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/Source/BasicMathFunctions/
Darm_elementwise_mul_acc_s16.c53 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()
Darm_elementwise_mul_s16_s8.c50 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/
Dtest_arm_depthwise_conv_s4.c35 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_depthwise_conv_s4_opt/
Dtest_arm_depthwise_conv_s4_opt.c33 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/Source/SoftmaxFunctions/
Darm_softmax_s8_s16.c49 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/Source/ConvolutionFunctions/
Darm_depthwise_conv_wrapper_s16.c60 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()
Darm_depthwise_conv_wrapper_s8.c59 int8_t *output) in arm_depthwise_conv_wrapper_s8() argument
79 output); in arm_depthwise_conv_wrapper_s8()
94 output); in arm_depthwise_conv_wrapper_s8()
109 output); in arm_depthwise_conv_wrapper_s8()
Darm_depthwise_conv_wrapper_s4.c59 int8_t *output) in arm_depthwise_conv_wrapper_s4() argument
75 output); in arm_depthwise_conv_wrapper_s4()
89 output); in arm_depthwise_conv_wrapper_s4()
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_elementwise_mul_s8/
Dtest_arm_elementwise_mul_s8.c28 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/Tests/UnitTest/TestCases/test_arm_convolve_s16/
Dtest_arm_convolve_s16.c42 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_convolve_s8/
Dtest_arm_convolve_s8.c42 int8_t output[BASIC_DST_SIZE] = {0}; in basic_arm_convolve_s8() local
97 output); in basic_arm_convolve_s8()
106 TEST_ASSERT_TRUE(validate(output, output_ref, output_ref_size)); in basic_arm_convolve_s8()
107 memset(output, 0, sizeof(output)); in basic_arm_convolve_s8()
123 output); in basic_arm_convolve_s8()
131 TEST_ASSERT_TRUE(validate(output, output_ref, output_ref_size)); in basic_arm_convolve_s8()
137 int8_t output[STRIDE2PAD1_DST_SIZE] = {0}; in stride2pad1_arm_convolve_s8() local
192 output); in stride2pad1_arm_convolve_s8()
200 TEST_ASSERT_TRUE(validate(output, output_ref, output_ref_size)); in stride2pad1_arm_convolve_s8()
201 memset(output, 0, sizeof(output)); in stride2pad1_arm_convolve_s8()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_convolve_1_x_n_s8/
Dtest_arm_convolve_1_x_n_s8.c38 int8_t output[CONV_1_X_N_1_DST_SIZE] = {0}; in conv_1_x_n_1_arm_convolve_s8() local
93 output); in conv_1_x_n_1_arm_convolve_s8()
101 TEST_ASSERT_TRUE(validate(output, output_ref, output_ref_size)); in conv_1_x_n_1_arm_convolve_s8()
102 memset(output, 0, sizeof(output)); in conv_1_x_n_1_arm_convolve_s8()
117 output); in conv_1_x_n_1_arm_convolve_s8()
124 TEST_ASSERT_TRUE(validate(output, output_ref, output_ref_size)); in conv_1_x_n_1_arm_convolve_s8()
130 int8_t output[CONV_1_X_N_2_DST_SIZE] = {0}; in conv_1_x_n_2_arm_convolve_s8() local
185 output); in conv_1_x_n_2_arm_convolve_s8()
193 memset(output, 0, sizeof(output)); in conv_1_x_n_2_arm_convolve_s8()
208 output); in conv_1_x_n_2_arm_convolve_s8()
[all …]
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_elementwise_add_s16/
Dtest_arm_elementwise_add_s16.c29 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_convolve_s4/
Dtest_arm_convolve_s4.c48 int8_t output[BASIC_INT4_DST_SIZE] = {0}; in basic_arm_convolve_s4() local
102 output); in basic_arm_convolve_s4()
111 TEST_ASSERT_TRUE(validate(output, output_ref, output_ref_size)); in basic_arm_convolve_s4()
112 memset(output, 0, sizeof(output)); in basic_arm_convolve_s4()
128 output); in basic_arm_convolve_s4()
136 TEST_ASSERT_TRUE(validate(output, output_ref, output_ref_size)); in basic_arm_convolve_s4()
142 int8_t output[BASIC_2_INT4_DST_SIZE] = {0}; in basic_2_arm_convolve_s4() local
196 output); in basic_2_arm_convolve_s4()
205 TEST_ASSERT_TRUE(validate(output, output_ref, output_ref_size)); in basic_2_arm_convolve_s4()
206 memset(output, 0, sizeof(output)); in basic_2_arm_convolve_s4()
[all …]

1234