#include "../TfLiteParser.hpp"
#include <doctest/doctest.h>
Go to the source code of this file.
|
| TEST_SUITE ("TensorflowLiteParser_OutputShapeOfSqueeze") |
|
◆ TEST_SUITE()
TEST_SUITE |
( |
"TensorflowLiteParser_OutputShapeOfSqueeze" |
| ) |
|
Definition at line 10 of file OutputShapeOfSqueeze.cpp.
References armnn::Float32, TensorInfo::GetNumDimensions(), TensorInfo::GetNumElements(), TensorInfo::GetShape(), and TEST_CASE_FIXTURE().
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")
unsigned int GetNumDimensions() const
unsigned int GetNumElements() const