Home
last modified time | relevance | path

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

123

/cmsis-nn-3.6.0-3.5.0/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-3.6.0-3.5.0/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-3.6.0-3.5.0/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-3.6.0-3.5.0/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-3.6.0-3.5.0/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-3.6.0-3.5.0/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-3.6.0-3.5.0/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()
Darm_softmax_u8.c50 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()
/cmsis-nn-3.6.0-3.5.0/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-3.6.0-3.5.0/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()
/cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_convolve_1_x_n_s8/
Dtest_arm_convolve_1_x_n_s8.c34 int8_t output[CONV_1_X_N_1_DST_SIZE] = {0}; in conv_1_x_n_1_arm_convolve_s8() local
88 output); in conv_1_x_n_1_arm_convolve_s8()
96 TEST_ASSERT_TRUE(validate(output, output_ref, output_ref_size)); in conv_1_x_n_1_arm_convolve_s8()
97 memset(output, 0, sizeof(output)); in conv_1_x_n_1_arm_convolve_s8()
112 output); in conv_1_x_n_1_arm_convolve_s8()
119 TEST_ASSERT_TRUE(validate(output, output_ref, output_ref_size)); in conv_1_x_n_1_arm_convolve_s8()
125 int8_t output[CONV_1_X_N_2_DST_SIZE] = {0}; in conv_1_x_n_2_arm_convolve_s8() local
179 output); in conv_1_x_n_2_arm_convolve_s8()
187 memset(output, 0, sizeof(output)); in conv_1_x_n_2_arm_convolve_s8()
202 output); in conv_1_x_n_2_arm_convolve_s8()
[all …]
/cmsis-nn-3.6.0-3.5.0/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
96 output); in basic_arm_convolve_s8()
105 TEST_ASSERT_TRUE(validate(output, output_ref, output_ref_size)); in basic_arm_convolve_s8()
106 memset(output, 0, sizeof(output)); in basic_arm_convolve_s8()
122 output); in basic_arm_convolve_s8()
130 TEST_ASSERT_TRUE(validate(output, output_ref, output_ref_size)); in basic_arm_convolve_s8()
136 int8_t output[STRIDE2PAD1_DST_SIZE] = {0}; in stride2pad1_arm_convolve_s8() local
190 output); in stride2pad1_arm_convolve_s8()
198 TEST_ASSERT_TRUE(validate(output, output_ref, output_ref_size)); in stride2pad1_arm_convolve_s8()
199 memset(output, 0, sizeof(output)); in stride2pad1_arm_convolve_s8()
[all …]
/cmsis-nn-3.6.0-3.5.0/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-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_depthwise_conv_fast_s16/
Dtest_arm_depthwise_conv_fast_s16.c37 int16_t output[DW_INT16XINT8_FAST_DST_SIZE] = {0}; in dw_int16xint8_fast_arm_depthwise_conv_fast_s16() local
92 output); in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
101 TEST_ASSERT_TRUE(validate_s16(output, output_ref, output_ref_size)); in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
102 memset(output, 0, sizeof(output)); in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
120 output); in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
128 TEST_ASSERT_TRUE(validate_s16(output, output_ref, output_ref_size)); in dw_int16xint8_fast_arm_depthwise_conv_fast_s16()
134 int16_t output[DW_INT16XINT8_FAST_SPILL_DST_SIZE] = {0}; in dw_int16xint8_fast_spill_arm_depthwise_conv_fast_s16() local
189 output); in dw_int16xint8_fast_spill_arm_depthwise_conv_fast_s16()
197 TEST_ASSERT_TRUE(validate_s16(output, output_ref, output_ref_size)); in dw_int16xint8_fast_spill_arm_depthwise_conv_fast_s16()
198 memset(output, 0, sizeof(output)); in dw_int16xint8_fast_spill_arm_depthwise_conv_fast_s16()
[all …]
/cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_max_pool_s8/
Dtest_arm_max_pool_s8.c37 int8_t output[MAXPOOLING_DST_SIZE] = {0}; in maxpooling_arm_max_pool_s8() local
68 … arm_max_pool_s8(&ctx, &pool_params, &input_dims, input_data, &filter_dims, &output_dims, output); in maxpooling_arm_max_pool_s8()
71 TEST_ASSERT_TRUE(validate(output, maxpooling_output_ref, MAXPOOLING_DST_SIZE)); in maxpooling_arm_max_pool_s8()
78 int8_t output[MAXPOOLING_1_DST_SIZE] = {0}; in maxpooling_1_arm_max_pool_s8() local
109 … arm_max_pool_s8(&ctx, &pool_params, &input_dims, input_data, &filter_dims, &output_dims, output); in maxpooling_1_arm_max_pool_s8()
112 TEST_ASSERT_TRUE(validate(output, maxpooling_1_output_ref, MAXPOOLING_1_DST_SIZE)); in maxpooling_1_arm_max_pool_s8()
119 int8_t output[MAXPOOLING_2_DST_SIZE] = {0}; in maxpooling_2_arm_max_pool_s8() local
150 … arm_max_pool_s8(&ctx, &pool_params, &input_dims, input_data, &filter_dims, &output_dims, output); in maxpooling_2_arm_max_pool_s8()
153 TEST_ASSERT_TRUE(validate(output, maxpooling_2_output_ref, MAXPOOLING_2_DST_SIZE)); in maxpooling_2_arm_max_pool_s8()
160 int8_t output[MAXPOOLING_3_DST_SIZE] = {0}; in maxpooling_3_arm_max_pool_s8() local
[all …]
/cmsis-nn-3.6.0-3.5.0/Source/FullyConnectedFunctions/
Darm_fully_connected_s16.c59 int16_t *output) in arm_fully_connected_s16() argument
74 output, in arm_fully_connected_s16()
82 output += output_dims->c; in arm_fully_connected_s16()
Darm_fully_connected_s8.c60 int8_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-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_convolve_s16/
Dtest_arm_convolve_s16.c34 int16_t output[INT16XINT8_DST_SIZE] = {0}; in int16xint8_arm_convolve_s16() local
87 output); in int16xint8_arm_convolve_s16()
95 TEST_ASSERT_TRUE(validate_s16(output, output_ref, output_ref_size)); in int16xint8_arm_convolve_s16()
96 memset(output, 0, sizeof(output)); in int16xint8_arm_convolve_s16()
111 output); in int16xint8_arm_convolve_s16()
119 TEST_ASSERT_TRUE(validate_s16(output, output_ref, output_ref_size)); in int16xint8_arm_convolve_s16()
124 int16_t output[REQUANTIZE_S64_DST_SIZE] = {0}; in requantize_s64_arm_convolve_s16() local
177 output); in requantize_s64_arm_convolve_s16()
184 TEST_ASSERT_TRUE(validate_s16(output, output_ref, output_ref_size)); in requantize_s64_arm_convolve_s16()
185 memset(output, 0, sizeof(output)); in requantize_s64_arm_convolve_s16()
[all …]
/cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_max_pool_s16/
Dtest_arm_max_pool_s16.c32 int16_t output[MAXPOOL_INT16_DST_SIZE] = {0}; in maxpool_int16_arm_max_pool_s16() local
63 … arm_max_pool_s16(&ctx, &pool_params, &input_dims, input_data, &filter_dims, &output_dims, output); in maxpool_int16_arm_max_pool_s16()
66 TEST_ASSERT_TRUE(validate_s16(output, maxpool_int16_output_ref, MAXPOOL_INT16_DST_SIZE)); in maxpool_int16_arm_max_pool_s16()
73 int16_t output[MAXPOOL_INT16_1_DST_SIZE] = {0}; in maxpool_int16_1_arm_max_pool_s16() local
104 … arm_max_pool_s16(&ctx, &pool_params, &input_dims, input_data, &filter_dims, &output_dims, output); in maxpool_int16_1_arm_max_pool_s16()
107 … TEST_ASSERT_TRUE(validate_s16(output, maxpool_int16_1_output_ref, MAXPOOL_INT16_1_DST_SIZE)); in maxpool_int16_1_arm_max_pool_s16()
114 int16_t output[MAXPOOL_INT16_2_DST_SIZE] = {0}; in maxpool_int16_2_arm_max_pool_s16() local
145 … arm_max_pool_s16(&ctx, &pool_params, &input_dims, input_data, &filter_dims, &output_dims, output); in maxpool_int16_2_arm_max_pool_s16()
148 … TEST_ASSERT_TRUE(validate_s16(output, maxpool_int16_2_output_ref, MAXPOOL_INT16_2_DST_SIZE)); in maxpool_int16_2_arm_max_pool_s16()
/cmsis-nn-3.6.0-3.5.0/Source/BasicMathFunctions/
Darm_elementwise_mul_s16_s8.c50 int8_t *output, in arm_elementwise_mul_s16_s8() argument
75 vstrbq_p_s32(output, res_0, pred); in arm_elementwise_mul_s16_s8()
79 output += 4; in arm_elementwise_mul_s16_s8()
101 arm_nn_write_s8x2_ia(&output, mul); in arm_elementwise_mul_s16_s8()
113 output[i] = (int8_t)mul_res; in arm_elementwise_mul_s16_s8()
/cmsis-nn-3.6.0-3.5.0/Tests/UnitTest/TestCases/test_arm_elementwise_add_s8/
Dtest_arm_elementwise_add_s8.c28 int8_t output[ADD_DST_SIZE] = {0}; in add_arm_elementwise_add_s8() local
58 output, in add_arm_elementwise_add_s8()
67 TEST_ASSERT_TRUE(validate(output, add_output_ref, ADD_DST_SIZE)); in add_arm_elementwise_add_s8()

123