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 } {
20 ~TfLiteParserFixture() { }
30 std::vector<uint32_t> squeezeDims = { };
41 std::vector<uint32_t> squeezeDims = { 1, 2 };
52 std::vector<uint32_t> squeezeDims = { 1, 3 };
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
BOOST_FIXTURE_TEST_CASE(EmptySqueezeDims_OutputWithAllDimensionsSqueezed, TfLiteParserFixture)
const TensorShape & GetShape() const
static ITfLiteParserPtr Create(const armnn::Optional< TfLiteParserOptions > &options=armnn::EmptyOptional())
BOOST_AUTO_TEST_SUITE_END()
static armnn::TensorInfo OutputShapeOfSqueeze(const std::vector< uint32_t > &squeezeDims, const armnn::TensorInfo &inputTensorInfo)
unsigned int GetNumDimensions() const
unsigned int GetNumElements() const