6 #include <boost/test/unit_test.hpp> 8 #include "../TfLiteParser.hpp" 17 explicit MinimumFixture(
const std::string & inputShape1,
18 const std::string & inputShape2,
19 const std::string & outputShape)
24 "operator_codes": [ { "builtin_code": "MINIMUM" } ], 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 MinimumFixture4D : MinimumFixture
88 MinimumFixture4D() : MinimumFixture(
"[ 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", { 0.0f, 0.0f, 0.0f,
104 9.0f, 9.0f, 9.0f }}},
105 {{
"outputTensor", { 0.0f, 0.0f, 0.0f,
108 9.0f, 9.0f, 9.0f }}});
111 struct MinimumBroadcastFixture4D : MinimumFixture
113 MinimumBroadcastFixture4D() : MinimumFixture(
"[ 1, 1, 2, 1 ]",
120 RunTest<4, armnn::DataType::Float32>(
122 {{
"inputTensor1", { 2.0f,
124 {
"inputTensor2", { 1.0f, 2.0f, 3.0f,
125 4.0f, 5.0f, 6.0f }}},
126 {{
"outputTensor", { 1.0f, 2.0f, 2.0f,
129 4.0f, 4.0f, 4.0f }}});
132 struct MinimumBroadcastFixture4D1D : MinimumFixture
134 MinimumBroadcastFixture4D1D() : MinimumFixture(
"[ 1, 2, 2, 3 ]",
141 RunTest<4, armnn::DataType::Float32>(
143 {{
"inputTensor1", { 0.0f, 1.0f, 2.0f,
146 9.0f, 10.0f, 11.0f }},
147 {
"inputTensor2", { 5.0f }}},
148 {{
"outputTensor", { 0.0f, 1.0f, 2.0f,
151 5.0f, 5.0f, 5.0f }}});
154 struct MinimumBroadcastFixture1D4D : MinimumFixture
156 MinimumBroadcastFixture1D4D() : MinimumFixture(
"[ 3 ]",
163 RunTest<4, armnn::DataType::Float32>(
165 {{
"inputTensor1", { 5.0f, 6.0f, 7.0f }},
166 {
"inputTensor2", { 0.0f, 1.0f, 2.0f,
169 9.0f, 10.0f, 11.0f }}},
170 {{
"outputTensor", { 0.0f, 1.0f, 2.0f,
173 5.0f, 6.0f, 7.0f }}});
178 explicit MinimumBroadcastFixture2D0D()
183 "operator_codes": [ { "builtin_code": "MINIMUM" } ], 230 "custom_options_format": "FLEXBUFFERS" 237 { "data": [ 0, 0, 0, 64 ] } 247 RunTest<2, armnn::DataType::Float32>( 249 {{"input0", { 1.0f, 5.0f }}},
250 {{
"output", { 1.0f, 2.0f }}});
BOOST_AUTO_TEST_SUITE_END()
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
BOOST_FIXTURE_TEST_CASE(ParseMinimum4D, MinimumFixture4D)