1 //
2 // Copyright © 2020, 2023 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5
6 #include "SplitTestHelper.hpp"
7
8 #include <armnn_delegate.hpp>
9
10 #include <flatbuffers/flatbuffers.h>
11 #include <schema_generated.h>
12
13 #include <doctest/doctest.h>
14
15 namespace armnnDelegate
16 {
17
18 // SPLIT Operator
SplitUint8Test(std::vector<armnn::BackendId> & backends)19 void SplitUint8Test(std::vector<armnn::BackendId>& backends)
20 {
21 std::vector<int32_t> axisShape { 1 };
22 std::vector<int32_t> inputShape { 2, 2, 2, 2} ;
23 std::vector<int32_t> outputShape0 { 2, 2, 2, 1 };
24 std::vector<int32_t> outputShape1 { 2, 2, 2, 1 };
25 std::vector<std::vector<int32_t>> outputShapes{ outputShape0, outputShape1 };
26
27 std::vector<int32_t> axisData { 3 }; // Axis
28 std::vector<uint8_t> inputValues { 1, 2, 3, 4, 5, 6, 7, 8,
29 9, 10, 11, 12, 13, 14, 15, 16 }; // Input
30
31
32 std::vector<uint8_t> expectedOutputValues0 { 1, 3, 5, 7, 9, 11, 13, 15 };
33 std::vector<uint8_t> expectedOutputValues1 { 2, 4, 6, 8, 10, 12, 14, 16 };
34 std::vector<std::vector<uint8_t>> expectedOutputValues{ expectedOutputValues0, expectedOutputValues1 };
35
36 int32_t numSplits = 2;
37
38 SplitTest<uint8_t>(::tflite::TensorType_UINT8,
39 backends,
40 axisShape,
41 inputShape,
42 outputShapes,
43 axisData,
44 inputValues,
45 expectedOutputValues,
46 numSplits);
47 }
48
SplitFp32Test(std::vector<armnn::BackendId> & backends)49 void SplitFp32Test(std::vector<armnn::BackendId>& backends)
50 {
51 std::vector<int32_t> axisShape { 1 };
52 std::vector<int32_t> inputShape { 2, 2, 2, 2 };
53 std::vector<int32_t> outputShape0 { 2, 1, 2, 2 };
54 std::vector<int32_t> outputShape1 { 2, 1, 2, 2 };
55 std::vector<std::vector<int32_t>> outputShapes{ outputShape0, outputShape1 };
56
57 std::vector<int32_t> axisData { 1 }; // Axis
58 std::vector<float> inputValues { 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f,
59 9.0f, 10.0f, 11.0f, 12.0f, 13.0f, 14.0f, 15.0f, 16.0f }; // Input
60
61
62 std::vector<float> expectedOutputValues0 { 1.0f, 2.0f, 3.0f, 4.0f, 9.0f, 10.0f, 11.0f, 12.0f };
63 std::vector<float> expectedOutputValues1 { 5.0f, 6.0f, 7.0f, 8.0f, 13.0f, 14.0f, 15.0f, 16.0f };
64 std::vector<std::vector<float>> expectedOutputValues{ expectedOutputValues0, expectedOutputValues1 };
65
66 int32_t numSplits = 2;
67
68 SplitTest<float>(::tflite::TensorType_FLOAT32,
69 backends,
70 axisShape,
71 inputShape,
72 outputShapes,
73 axisData,
74 inputValues,
75 expectedOutputValues,
76 numSplits);
77 }
78
79 // SPLIT Test Suite
80 TEST_SUITE("SPLIT_CpuRefTests")
81 {
82
83 TEST_CASE ("SPLIT_Uint8_CpuRef_Test")
84 {
85 std::vector<armnn::BackendId> backends = {armnn::Compute::CpuRef};
86 SplitUint8Test(backends);
87 }
88
89 TEST_CASE ("SPLIT_Fp32_CpuRef_Test")
90 {
91 std::vector<armnn::BackendId> backends = {armnn::Compute::CpuRef};
92 SplitFp32Test(backends);
93 }
94
95 }
96
97 TEST_SUITE("SPLIT_CpuAccTests")
98 {
99
100 TEST_CASE ("SPLIT_Uint8_CpuAcc_Test")
101 {
102 std::vector<armnn::BackendId> backends = {armnn::Compute::CpuAcc};
103 SplitUint8Test(backends);
104 }
105
106 TEST_CASE ("SPLIT_Fp32_CpuAcc_Test")
107 {
108 std::vector<armnn::BackendId> backends = {armnn::Compute::CpuAcc};
109 SplitFp32Test(backends);
110 }
111
112 }
113
114 TEST_SUITE("SPLIT_GpuAccTests")
115 {
116
117 TEST_CASE ("SPLIT_Uint8_GpuAcc_Test")
118 {
119 std::vector<armnn::BackendId> backends = {armnn::Compute::GpuAcc};
120 SplitUint8Test(backends);
121 }
122
123 TEST_CASE ("SPLIT_Fp32_GpuAcc_Test")
124 {
125 std::vector<armnn::BackendId> backends = {armnn::Compute::GpuAcc};
126 SplitFp32Test(backends);
127 }
128
129 }
130 // End of SPLIT Test Suite
131
132 // SPLIT_V Operator
SplitVUint8Test(std::vector<armnn::BackendId> & backends)133 void SplitVUint8Test(std::vector<armnn::BackendId>& backends)
134 {
135 std::vector<int32_t> axisShape { 1 };
136 std::vector<int32_t> inputShape { 2, 4, 2, 2 };
137 std::vector<int32_t> splitsShape { 2 };
138 std::vector<int32_t> outputShape0 { 2, 3, 2, 2 };
139 std::vector<int32_t> outputShape1 { 2, 1, 2, 2 };
140 std::vector<std::vector<int32_t>> outputShapes{ outputShape0, outputShape1 };
141
142 std::vector<int32_t> axisData { 1 }; // Axis
143 std::vector<int32_t> splitsData { 3, 1 }; // Splits
144 std::vector<uint8_t> inputValues { 1, 2, 3, 4, 5, 6, 7, 8,
145 9, 10, 11, 12, 13, 14, 15, 16,
146 17, 18, 19, 20, 21, 22, 23, 24,
147 25, 26, 27, 28, 29, 30, 31, 32 }; // Input
148
149
150 std::vector<uint8_t> expectedOutputValues0 { 1, 2, 3, 4, 5, 6, 7, 8,
151 9, 10, 11, 12, 17, 18, 19, 20,
152 21, 22, 23, 24, 25, 26, 27, 28 };
153 std::vector<uint8_t> expectedOutputValues1 { 13, 14, 15, 16, 29, 30, 31, 32 };
154 std::vector<std::vector<uint8_t>> expectedOutputValues{ expectedOutputValues0, expectedOutputValues1 };
155
156 int32_t numSplits = 2;
157
158 SplitVTest<uint8_t>(::tflite::TensorType_UINT8,
159 backends,
160 inputShape,
161 splitsShape,
162 axisShape,
163 outputShapes,
164 inputValues,
165 splitsData,
166 axisData,
167 expectedOutputValues,
168 numSplits);
169 }
170
SplitVFp32Test(std::vector<armnn::BackendId> & backends)171 void SplitVFp32Test(std::vector<armnn::BackendId>& backends)
172 {
173 std::vector<int32_t> axisShape { 1 };
174 std::vector<int32_t> inputShape { 2, 4, 2, 2 };
175 std::vector<int32_t> splitsShape { 2 };
176 std::vector<int32_t> outputShape0 { 2, 3, 2, 2 };
177 std::vector<int32_t> outputShape1 { 2, 1, 2, 2 };
178 std::vector<std::vector<int32_t>> outputShapes{ outputShape0, outputShape1 };
179
180 std::vector<int32_t> axisData { 1 }; // Axis
181 std::vector<int32_t> splitsData { 3, 1 }; // Splits
182 std::vector<float> inputValues { 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f,
183 9.0f, 10.0f, 11.0f, 12.0f, 13.0f, 14.0f, 15.0f, 16.0f,
184 17.0f, 18.0f, 19.0f, 20.0f, 21.0f, 22.0f, 23.0f, 24.0f,
185 25.0f, 26.0f, 27.0f, 28.0f, 29.0f, 30.0f, 31.0f, 32.0f }; // Input
186
187
188 std::vector<float> expectedOutputValues0 { 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f,
189 9.0f, 10.0f, 11.0f, 12.0f, 17.0f, 18.0f, 19.0f, 20.0f,
190 21.0f, 22.0f, 23.0f, 24.0f, 25.0f, 26.0f, 27.0f, 28.0f };
191 std::vector<float> expectedOutputValues1 { 13.0f, 14.0f, 15.0f, 16.0f, 29.0f, 30.0f, 31.0f, 32.0f };
192 std::vector<std::vector<float>> expectedOutputValues{ expectedOutputValues0, expectedOutputValues1 };
193
194 int32_t numSplits = 2;
195
196 SplitVTest<float>(::tflite::TensorType_FLOAT32,
197 backends,
198 inputShape,
199 splitsShape,
200 axisShape,
201 outputShapes,
202 inputValues,
203 splitsData,
204 axisData,
205 expectedOutputValues,
206 numSplits);
207 }
208
209 // SPLIT_V Test Suite
210 TEST_SUITE("SPLIT_V_CpuRefTests")
211 {
212
213 TEST_CASE ("SPLIT_V_Uint8_CpuRef_Test")
214 {
215 std::vector<armnn::BackendId> backends = {armnn::Compute::CpuRef};
216 SplitVUint8Test(backends);
217 }
218
219 TEST_CASE ("SPLIT_V_Fp32_CpuRef_Test")
220 {
221 std::vector<armnn::BackendId> backends = {armnn::Compute::CpuRef};
222 SplitVFp32Test(backends);
223 }
224
225 }
226
227 TEST_SUITE("SPLIT_V_CpuAccTests")
228 {
229
230 TEST_CASE ("SPLIT_V_Uint8_CpuAcc_Test")
231 {
232 std::vector<armnn::BackendId> backends = {armnn::Compute::CpuAcc};
233 SplitVUint8Test(backends);
234 }
235
236 TEST_CASE ("SPLIT_V_Fp32_CpuAcc_Test")
237 {
238 std::vector<armnn::BackendId> backends = {armnn::Compute::CpuAcc};
239 SplitVFp32Test(backends);
240 }
241
242 }
243
244 TEST_SUITE("SPLIT_V_GpuAccTests")
245 {
246
247 TEST_CASE ("SPLIT_V_Uint8_GpuAcc_Test")
248 {
249 std::vector<armnn::BackendId> backends = {armnn::Compute::GpuAcc};
250 SplitVUint8Test(backends);
251 }
252
253 TEST_CASE ("SPLIT_V_Fp32_GpuAcc_Test")
254 {
255 std::vector<armnn::BackendId> backends = {armnn::Compute::GpuAcc};
256 SplitVFp32Test(backends);
257 }
258
259 }
260 // End of SPLIT_V Test Suite
261
262 } // namespace armnnDelegate