1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
//
// Copyright © 2021, 2023-2024 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
#include "UnpackTestHelper.hpp"
#include <doctest/doctest.h>
namespace armnnDelegate
{
template <typename T>
void UnpackAxis0Num4Test(tflite::TensorType tensorType)
{
std::vector<int32_t> inputShape { 4, 1, 6 };
std::vector<int32_t> expectedOutputShape { 1, 6 };
std::vector<T> inputValues { 1, 2, 3, 4, 5, 6,
7, 8, 9, 10, 11, 12,
13, 14, 15, 16, 17, 18,
19, 20, 21, 22, 23, 24 };
std::vector<T> expectedOutputValues0 { 1, 2, 3, 4, 5, 6 };
std::vector<T> expectedOutputValues1 { 7, 8, 9, 10, 11, 12 };
std::vector<T> expectedOutputValues2 { 13, 14, 15, 16, 17, 18 };
std::vector<T> expectedOutputValues3 { 19, 20, 21, 22, 23, 24 };
std::vector<std::vector<T>> expectedOutputValues{ expectedOutputValues0,
expectedOutputValues1,
expectedOutputValues2,
expectedOutputValues3 };
UnpackTest<T>(tflite::BuiltinOperator_UNPACK,
tensorType,
inputShape,
expectedOutputShape,
inputValues,
expectedOutputValues,
{},
0);
}
template <typename T>
void UnpackAxis2Num6Test(tflite::TensorType tensorType)
{
std::vector<int32_t> inputShape { 4, 1, 6 };
std::vector<int32_t> expectedOutputShape { 4, 1 };
std::vector<T> inputValues { 1, 2, 3, 4, 5, 6,
7, 8, 9, 10, 11, 12,
13, 14, 15, 16, 17, 18,
19, 20, 21, 22, 23, 24 };
std::vector<T> expectedOutputValues0 { 1, 7, 13, 19 };
std::vector<T> expectedOutputValues1 { 2, 8, 14, 20 };
std::vector<T> expectedOutputValues2 { 3, 9, 15, 21 };
std::vector<T> expectedOutputValues3 { 4, 10, 16, 22 };
std::vector<T> expectedOutputValues4 { 5, 11, 17, 23 };
std::vector<T> expectedOutputValues5 { 6, 12, 18, 24 };
std::vector<std::vector<T>> expectedOutputValues{ expectedOutputValues0,
expectedOutputValues1,
expectedOutputValues2,
expectedOutputValues3,
expectedOutputValues4,
expectedOutputValues5 };
UnpackTest<T>(tflite::BuiltinOperator_UNPACK,
tensorType,
inputShape,
expectedOutputShape,
inputValues,
expectedOutputValues,
{},
2);
}
TEST_SUITE("UnpackTests")
{
// Fp32
TEST_CASE ("Unpack_Fp32_Axis0_Num4_Test")
{
UnpackAxis0Num4Test<float>(tflite::TensorType_FLOAT32);
}
TEST_CASE ("Unpack_Fp32_Axis2_Num6_Test")
{
UnpackAxis2Num6Test<float>(tflite::TensorType_FLOAT32);
}
// Uint8
TEST_CASE ("Unpack_Uint8_Axis0_Num4_Test")
{
UnpackAxis0Num4Test<uint8_t>(tflite::TensorType_UINT8);
}
TEST_CASE ("Unpack_Uint8_Axis2_Num6_Test")
{
UnpackAxis2Num6Test<uint8_t>(tflite::TensorType_UINT8);
}
}
// End of Unpack Test Suite
} // namespace armnnDelegate
|
