1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5
6 #include <boost/test/unit_test.hpp>
7 #include "ParserFlatbuffersFixture.hpp"
8 #include "../TfLiteParser.hpp"
9
10 #include <string>
11 #include <iostream>
12
13 BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
14
15 struct MultiplicationFixture : public ParserFlatbuffersFixture
16 {
MultiplicationFixtureMultiplicationFixture17 explicit MultiplicationFixture(const std::string & inputShape1,
18 const std::string & inputShape2,
19 const std::string & outputShape,
20 const std::string & activation="NONE")
21 {
22 m_JsonString = R"(
23 {
24 "version": 3,
25 "operator_codes": [ { "builtin_code": "MUL" } ],
26 "subgraphs": [ {
27 "tensors": [
28 {
29 "shape": )" + inputShape1 + R"(,
30 "type": "FLOAT32",
31 "buffer": 0,
32 "name": "inputTensor1",
33 "quantization": {
34 "min": [ 0.0 ],
35 "max": [ 255.0 ],
36 "scale": [ 1.0 ],
37 "zero_point": [ 0 ],
38 }
39 },
40 {
41 "shape": )" + inputShape2 + R"(,
42 "type": "FLOAT32",
43 "buffer": 1,
44 "name": "inputTensor2",
45 "quantization": {
46 "min": [ 0.0 ],
47 "max": [ 255.0 ],
48 "scale": [ 1.0 ],
49 "zero_point": [ 0 ],
50 }
51 },
52 {
53 "shape": )" + outputShape + R"( ,
54 "type": "FLOAT32",
55 "buffer": 2,
56 "name": "outputTensor",
57 "quantization": {
58 "min": [ 0.0 ],
59 "max": [ 255.0 ],
60 "scale": [ 1.0 ],
61 "zero_point": [ 0 ],
62 }
63 }
64 ],
65 "inputs": [ 0, 1 ],
66 "outputs": [ 2 ],
67 "operators": [
68 {
69 "opcode_index": 0,
70 "inputs": [ 0, 1 ],
71 "outputs": [ 2 ],
72 "builtin_options_type": "MulOptions",
73 "builtin_options": {
74 "fused_activation_function": )" + activation + R"(
75 },
76 "custom_options_format": "FLEXBUFFERS"
77 }
78 ],
79 } ],
80 "buffers" : [
81 { },
82 { }
83 ]
84 }
85 )";
86 Setup();
87 }
88 };
89
90 struct SimpleMultiplicationFixture : public MultiplicationFixture
91 {
SimpleMultiplicationFixtureSimpleMultiplicationFixture92 SimpleMultiplicationFixture() : MultiplicationFixture("[ 1, 2, 2, 3 ]", "[ 1, 2, 2, 3 ]", "[ 1, 2, 2, 3 ]") {}
93 };
94
BOOST_FIXTURE_TEST_CASE(ParseMultiplication,SimpleMultiplicationFixture)95 BOOST_FIXTURE_TEST_CASE(ParseMultiplication, SimpleMultiplicationFixture)
96 {
97 using armnn::DataType;
98 RunTest<4, DataType::Float32>(0, {{ "inputTensor1", { 0.0f, 1.0f, 2.0f,
99 3.0f, 4.0f, 5.0f,
100 6.0f, 7.0f, 8.0f,
101 9.0f, 10.0f, 11.0f } },
102 { "inputTensor2", { 1.0f, 1.0f, 1.0f,
103 5.0f, 5.0f, 5.0f,
104 1.0f, 1.0f, 1.0f,
105 5.0f, 5.0f, 5.0f} } },
106 {{ "outputTensor", { 0.0f, 1.0f, 2.0f,
107 15.0f, 20.0f, 25.0f,
108 6.0f, 7.0f, 8.0f,
109 45.0f, 50.0f, 55.0f } } });
110 }
111
112 struct MultiplicationBroadcastFixture4D1D : public MultiplicationFixture
113 {
MultiplicationBroadcastFixture4D1DMultiplicationBroadcastFixture4D1D114 MultiplicationBroadcastFixture4D1D() : MultiplicationFixture("[ 1, 2, 2, 3 ]", "[ 1 ]", "[ 1, 2, 2, 3 ]") {}
115 };
116
BOOST_FIXTURE_TEST_CASE(ParseMultiplicationBroadcast4D1D,MultiplicationBroadcastFixture4D1D)117 BOOST_FIXTURE_TEST_CASE(ParseMultiplicationBroadcast4D1D, MultiplicationBroadcastFixture4D1D)
118 {
119 using armnn::DataType;
120 RunTest<4, DataType::Float32>(0, {{ "inputTensor1", { 0.0f, 1.0f, 2.0f,
121 3.0f, 4.0f, 5.0f,
122 6.0f, 7.0f, 8.0f,
123 9.0f, 10.0f, 11.0f } },
124 { "inputTensor2", { 5.0f } } },
125 {{ "outputTensor", { 0.0f, 5.0f, 10.0f,
126 15.0f, 20.0f, 25.0f,
127 30.0f, 35.0f, 40.0f,
128 45.0f, 50.0f, 55.0f } } });
129 }
130
131 struct MultiplicationBroadcastFixture1D4D : public MultiplicationFixture
132 {
MultiplicationBroadcastFixture1D4DMultiplicationBroadcastFixture1D4D133 MultiplicationBroadcastFixture1D4D() : MultiplicationFixture("[ 1 ]", "[ 1, 2, 2, 3 ]", "[ 1, 2, 2, 3 ]") {}
134 };
135
BOOST_FIXTURE_TEST_CASE(ParseMultiplicationBroadcast1D4D,MultiplicationBroadcastFixture1D4D)136 BOOST_FIXTURE_TEST_CASE(ParseMultiplicationBroadcast1D4D, MultiplicationBroadcastFixture1D4D)
137 {
138 using armnn::DataType;
139 RunTest<4, DataType::Float32>(0, {{ "inputTensor1", { 3.0f } },
140 { "inputTensor2", { 0.0f, 1.0f, 2.0f,
141 3.0f, 4.0f, 5.0f,
142 6.0f, 7.0f, 8.0f,
143 9.0f, 10.0f, 11.0f } } },
144 {{ "outputTensor", { 0.0f, 3.0f, 6.0f,
145 9.0f, 12.0f, 15.0f,
146 18.0f, 21.0f, 24.0f,
147 27.0f, 30.0f, 33.0f } } });
148 }
149
150 BOOST_AUTO_TEST_SUITE_END()
151