6 #include <boost/test/unit_test.hpp> 8 #include "../TfLiteParser.hpp" 17 explicit SqueezeFixture(
const std::string& inputShape,
18 const std::string& outputShape,
19 const std::string& squeezeDims)
24 "operator_codes": [ { "builtin_code": "SQUEEZE" } ], 29 "shape" : )" + inputShape + ",";
33 "name": "inputTensor", 43 "shape" : )" + outputShape; 47 "name": "outputTensor", 63 "builtin_options_type": "SqueezeOptions", 64 "builtin_options": {)"; 65 if (!squeezeDims.empty())
70 "custom_options_format": "FLEXBUFFERS" 74 "buffers" : [ {}, {} ] 80 struct SqueezeFixtureWithSqueezeDims : SqueezeFixture
82 SqueezeFixtureWithSqueezeDims() : SqueezeFixture(
"[ 1, 2, 2, 1 ]",
"[ 2, 2, 1 ]",
"[ 0, 1, 2 ]") {}
87 SetupSingleInputSingleOutput(
"inputTensor",
"outputTensor");
88 RunTest<3, armnn::DataType::QAsymmU8>(0, { 1, 2, 3, 4 }, { 1, 2, 3, 4 });
89 BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo(0,
"outputTensor").second.GetShape()
94 struct SqueezeFixtureWithoutSqueezeDims : SqueezeFixture
96 SqueezeFixtureWithoutSqueezeDims() : SqueezeFixture(
"[ 1, 2, 2, 1 ]",
"[ 2, 2 ]",
"") {}
101 SetupSingleInputSingleOutput(
"inputTensor",
"outputTensor");
102 RunTest<2, armnn::DataType::QAsymmU8>(0, { 1, 2, 3, 4 }, { 1, 2, 3, 4 });
103 BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo(0,
"outputTensor").second.GetShape()
107 struct SqueezeFixtureWithInvalidInput : SqueezeFixture
109 SqueezeFixtureWithInvalidInput() : SqueezeFixture(
"[ 1, 2, 2, 1, 2, 2 ]",
"[ 1, 2, 2, 1, 2 ]",
"[ ]") {}
115 BOOST_CHECK_THROW((SetupSingleInputSingleOutput(
"inputTensor",
"outputTensor")),
119 struct SqueezeFixtureWithSqueezeDimsSizeInvalid : SqueezeFixture
121 SqueezeFixtureWithSqueezeDimsSizeInvalid() : SqueezeFixture(
"[ 1, 2, 2, 1 ]",
123 "[ 1, 2, 2, 2, 2 ]") {}
128 BOOST_CHECK_THROW((SetupSingleInputSingleOutput(
"inputTensor",
"outputTensor")),
armnn::ParseException);
132 struct SqueezeFixtureWithNegativeSqueezeDims : SqueezeFixture
134 SqueezeFixtureWithNegativeSqueezeDims() : SqueezeFixture(
"[ 1, 2, 2, 1 ]",
141 BOOST_CHECK_THROW((SetupSingleInputSingleOutput(
"inputTensor",
"outputTensor")),
armnn::ParseException);
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
BOOST_FIXTURE_TEST_CASE(ParseSqueezeWithSqueezeDims, SqueezeFixtureWithSqueezeDims)
BOOST_AUTO_TEST_SUITE_END()
constexpr unsigned int MaxNumOfTensorDimensions