Reshape.cpp File Reference

#include <boost/test/unit_test.hpp>
#include "ParserFlatbuffersFixture.hpp"
#include "../TfLiteParser.hpp"
#include <string>
#include <iostream>

Go to the source code of this file.

Functions
	BOOST_FIXTURE_TEST_CASE (ParseReshapeWithReshapeDims, ReshapeFixtureWithReshapeDims)
	BOOST_FIXTURE_TEST_CASE (ParseReshapeWithReshapeDimsFlatten, ReshapeFixtureWithReshapeDimsFlatten)
	BOOST_FIXTURE_TEST_CASE (ParseReshapeWithReshapeDimsFlattenTwoDims, ReshapeFixtureWithReshapeDimsFlattenTwoDims)
	BOOST_FIXTURE_TEST_CASE (ParseReshapeWithReshapeDimsFlattenOneDim, ReshapeFixtureWithReshapeDimsFlattenOneDim)
	BOOST_FIXTURE_TEST_CASE (DynParseReshapeWithReshapeDimsFlattenOneDim, DynamicReshapeFixtureWithReshapeDimsFlattenOneDim)

Function Documentation

BOOST_FIXTURE_TEST_CASE() [1/5]

BOOST_FIXTURE_TEST_CASE	(	ParseReshapeWithReshapeDims	,
		ReshapeFixtureWithReshapeDims
	)

Definition at line 86 of file Reshape.cpp.

{
    SetupSingleInputSingleOutput("inputTensor", "outputTensor");
    RunTest<2, armnn::DataType::QAsymmU8>(0,
                                                 { 1, 2, 3, 4, 5, 6, 7, 8, 9 },
                                                 { 1, 2, 3, 4, 5, 6, 7, 8, 9 });
    BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo(0, "outputTensor").second.GetShape()
                == armnn::TensorShape({3,3})));
}

BOOST_FIXTURE_TEST_CASE() [2/5]

BOOST_FIXTURE_TEST_CASE	(	ParseReshapeWithReshapeDimsFlatten	,
		ReshapeFixtureWithReshapeDimsFlatten
	)

Definition at line 101 of file Reshape.cpp.

{
    SetupSingleInputSingleOutput("inputTensor", "outputTensor");
    RunTest<1, armnn::DataType::QAsymmU8>(0,
                                                 { 1, 2, 3, 4, 5, 6, 7, 8, 9 },
                                                 { 1, 2, 3, 4, 5, 6, 7, 8, 9 });
    BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo(0, "outputTensor").second.GetShape()
                == armnn::TensorShape({9})));
}

BOOST_FIXTURE_TEST_CASE() [3/5]

BOOST_FIXTURE_TEST_CASE	(	ParseReshapeWithReshapeDimsFlattenTwoDims	,
		ReshapeFixtureWithReshapeDimsFlattenTwoDims
	)

Definition at line 116 of file Reshape.cpp.

{
    SetupSingleInputSingleOutput("inputTensor", "outputTensor");
    RunTest<2, armnn::DataType::QAsymmU8>(0,
                                                 { 1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5, 6, 6, 6 },
                                                 { 1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5, 6, 6, 6 });
    BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo(0, "outputTensor").second.GetShape()
                == armnn::TensorShape({2,9})));
}

BOOST_FIXTURE_TEST_CASE() [4/5]

BOOST_FIXTURE_TEST_CASE	(	ParseReshapeWithReshapeDimsFlattenOneDim	,
		ReshapeFixtureWithReshapeDimsFlattenOneDim
	)

Definition at line 131 of file Reshape.cpp.

{
    SetupSingleInputSingleOutput("inputTensor", "outputTensor");
    RunTest<3, armnn::DataType::QAsymmU8>(0,
                                                 { 1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5, 6, 6, 6 },
                                                 { 1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5, 6, 6, 6 });
    BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo(0, "outputTensor").second.GetShape()
                == armnn::TensorShape({2,3,3})));
}

BOOST_FIXTURE_TEST_CASE() [5/5]

BOOST_FIXTURE_TEST_CASE	(	DynParseReshapeWithReshapeDimsFlattenOneDim	,
		DynamicReshapeFixtureWithReshapeDimsFlattenOneDim
	)

Definition at line 148 of file Reshape.cpp.

References BOOST_AUTO_TEST_SUITE_END(), and armnn::QAsymmU8.

{
    SetupSingleInputSingleOutput("inputTensor", "outputTensor");
     RunTest<3,
        armnn::DataType::QAsymmU8,
        armnn::DataType::QAsymmU8>(0,
                                   { { "inputTensor", {  1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5, 6, 6, 6 } } },
                                   { { "outputTensor", { 1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5, 6, 6, 6 } } },
                                   true);
}