6 #include "../TfLiteParser.hpp" 8 #include <doctest/doctest.h> 13 struct TfLiteParserFixture
17 unsigned int m_InputShape[4];
19 TfLiteParserFixture() : m_Parser( ), m_InputShape { 1, 2, 2, 1 } {}
20 ~TfLiteParserFixture() { }
24 TEST_CASE_FIXTURE(TfLiteParserFixture,
"EmptySqueezeDims_OutputWithAllDimensionsSqueezed")
27 std::vector<uint32_t> squeezeDims = { };
30 armnn::TensorInfo outputTensorInfo = m_Parser.OutputShapeOfSqueeze(squeezeDims, inputTensorInfo);
36 TEST_CASE_FIXTURE(TfLiteParserFixture,
"SqueezeDimsNotIncludingSizeOneDimensions_NoDimensionsSqueezedInOutput")
38 std::vector<uint32_t> squeezeDims = { 1, 2 };
41 armnn::TensorInfo outputTensorInfo = m_Parser.OutputShapeOfSqueeze(squeezeDims, inputTensorInfo);
47 TEST_CASE_FIXTURE(TfLiteParserFixture,
"SqueezeDimsRangePartial_OutputWithDimensionsWithinRangeSqueezed")
49 std::vector<uint32_t> squeezeDims = { 1, 3 };
52 armnn::TensorInfo outputTensorInfo = m_Parser.OutputShapeOfSqueeze(squeezeDims, inputTensorInfo);
const TensorShape & GetShape() const
TEST_CASE_FIXTURE(ClContextControlFixture, "CopyBetweenNeonAndGpu")
TEST_SUITE("TensorflowLiteParser_OutputShapeOfSqueeze")
unsigned int GetNumDimensions() const
unsigned int GetNumElements() const