6 #include <boost/test/unit_test.hpp> 7 #include "../TfLiteParser.hpp" 11 struct TfLiteParserFixture
15 unsigned int m_InputShape[4];
17 TfLiteParserFixture() : m_Parser( ), m_InputShape { 1, 2, 2, 1 } {}
18 ~TfLiteParserFixture() { }
28 std::vector<uint32_t> squeezeDims = { };
31 armnn::TensorInfo outputTensorInfo = m_Parser.OutputShapeOfSqueeze(squeezeDims, inputTensorInfo);
39 std::vector<uint32_t> squeezeDims = { 1, 2 };
42 armnn::TensorInfo outputTensorInfo = m_Parser.OutputShapeOfSqueeze(squeezeDims, inputTensorInfo);
50 std::vector<uint32_t> squeezeDims = { 1, 3 };
53 armnn::TensorInfo outputTensorInfo = m_Parser.OutputShapeOfSqueeze(squeezeDims, inputTensorInfo);
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
BOOST_FIXTURE_TEST_CASE(EmptySqueezeDims_OutputWithAllDimensionsSqueezed, TfLiteParserFixture)
const TensorShape & GetShape() const
BOOST_AUTO_TEST_SUITE_END()
unsigned int GetNumDimensions() const
unsigned int GetNumElements() const