6 #include <boost/test/unit_test.hpp> 8 #include "../TfLiteParser.hpp" 17 explicit MaximumFixture(
const std::string & inputShape1,
18 const std::string & inputShape2,
19 const std::string & outputShape)
24 "operator_codes": [ { "builtin_code": "MAXIMUM" } ], 28 "shape": )" + inputShape1 + R"(, 31 "name": "inputTensor1", 40 "shape": )" + inputShape2 + R"(, 43 "name": "inputTensor2", 52 "shape": )" + outputShape + R"( , 55 "name": "outputTensor", 71 "custom_options_format": "FLEXBUFFERS" 86 struct MaximumFixture4D4D : MaximumFixture
88 MaximumFixture4D4D() : MaximumFixture(
"[ 1, 2, 2, 3 ]",
95 RunTest<4, armnn::DataType::Float32>(
97 {{
"inputTensor1", { 0.0f, 1.0f, 2.0f,
100 9.0f, 10.0f, 11.0f }},
101 {
"inputTensor2", { 5.0f, 1.0f, 3.0f,
104 19.0f, 1.0f, 3.0f }}},
105 {{
"outputTensor", { 5.0f, 1.0f, 3.0f,
108 19.0f, 10.0f, 11.0f }}});
111 struct MaximumBroadcastFixture4D4D : MaximumFixture
113 MaximumBroadcastFixture4D4D() : MaximumFixture(
"[ 1, 1, 2, 1 ]",
120 RunTest<4, armnn::DataType::Float32>(
122 {{
"inputTensor1", { 2.0f, 4.0f }},
123 {
"inputTensor2", { 1.0f, 2.0f, 3.0f,
124 4.0f, 5.0f, 6.0f }}},
125 {{
"outputTensor", { 2.0f, 2.0f, 3.0f,
128 4.0f, 5.0f, 6.0f }}});
131 struct MaximumBroadcastFixture4D1D : MaximumFixture
133 MaximumBroadcastFixture4D1D() : MaximumFixture(
"[ 1, 2, 2, 3 ]",
140 RunTest<4, armnn::DataType::Float32>(
142 {{
"inputTensor1", { 0.0f, 1.0f, 2.0f,
145 9.0f, 10.0f, 11.0f }},
146 {
"inputTensor2", { 5.0f }}},
147 {{
"outputTensor", { 5.0f, 5.0f, 5.0f,
150 9.0f, 10.0f, 11.0f }}});
153 struct MaximumBroadcastFixture1D4D : MaximumFixture
155 MaximumBroadcastFixture1D4D() : MaximumFixture(
"[ 1 ]",
162 RunTest<4, armnn::DataType::Float32>(
164 {{
"inputTensor1", { 3.0f }},
165 {
"inputTensor2", { 0.0f, 1.0f, 2.0f,
168 9.0f, 10.0f, 11.0f }}},
169 {{
"outputTensor", { 3.0f, 3.0f, 3.0f,
172 9.0f, 10.0f, 11.0f }}});
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
BOOST_FIXTURE_TEST_CASE(ParseMaximum4D4D, MaximumFixture4D4D)
BOOST_AUTO_TEST_SUITE_END()