Split.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 (ParseAxisOneSplitTwoFloat32, SimpleSplitFixtureFloat32)
	BOOST_FIXTURE_TEST_CASE (ParseAxisThreeSplitTwoFloat32, SimpleSplitAxisThreeFixtureFloat32)
	BOOST_FIXTURE_TEST_CASE (SimpleSplit2DFloat32, SimpleSplit2DFixtureFloat32)
	BOOST_FIXTURE_TEST_CASE (SimpleSplit3DFloat32, SimpleSplit3DFixtureFloat32)
	BOOST_FIXTURE_TEST_CASE (ParseAxisOneSplitTwoUint8, SimpleSplitFixtureUint8)
	BOOST_FIXTURE_TEST_CASE (ParseAxisThreeSplitTwoUint8, SimpleSplitAxisThreeFixtureUint8)
	BOOST_FIXTURE_TEST_CASE (SimpleSplit2DUint8, SimpleSplit2DFixtureUint8)
	BOOST_FIXTURE_TEST_CASE (SimpleSplit3DUint8, SimpleSplit3DFixtureUint8)

Function Documentation

BOOST_FIXTURE_TEST_CASE() [1/8]

BOOST_FIXTURE_TEST_CASE	(	ParseAxisOneSplitTwoFloat32	,
		SimpleSplitFixtureFloat32
	)

Definition at line 111 of file Split.cpp.

{

    RunTest<4, armnn::DataType::Float32>(
        0,
        { {"inputTensor", { 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f,
                            9.0f, 10.0f, 11.0f, 12.0f, 13.0f, 14.0f, 15.0f, 16.0f } } },
        { {"outputTensor1", { 1.0f, 2.0f, 3.0f, 4.0f, 9.0f, 10.0f, 11.0f, 12.0f } },
          {"outputTensor2", { 5.0f, 6.0f, 7.0f, 8.0f, 13.0f, 14.0f, 15.0f, 16.0f } } });
}

BOOST_FIXTURE_TEST_CASE() [2/8]

BOOST_FIXTURE_TEST_CASE	(	ParseAxisThreeSplitTwoFloat32	,
		SimpleSplitAxisThreeFixtureFloat32
	)

Definition at line 129 of file Split.cpp.

{
    RunTest<4, armnn::DataType::Float32>(
        0,
        { {"inputTensor", { 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f,
                            9.0f, 10.0f, 11.0f, 12.0f, 13.0f, 14.0f, 15.0f, 16.0f } } },
        { {"outputTensor1", { 1.0f, 3.0f, 5.0f, 7.0f, 9.0f, 11.0f, 13.0f, 15.0f } },
          {"outputTensor2", { 2.0f, 4.0f, 6.0f, 8.0f, 10.0f, 12.0f, 14.0f, 16.0f } } } );
}

BOOST_FIXTURE_TEST_CASE() [3/8]

BOOST_FIXTURE_TEST_CASE	(	SimpleSplit2DFloat32	,
		SimpleSplit2DFixtureFloat32
	)

Definition at line 146 of file Split.cpp.

{
    RunTest<2, armnn::DataType::Float32>(
        0,
        { {"inputTensor", { 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f } } },
        { {"outputTensor1", { 1.0f, 2.0f, 3.0f, 4.0f } },
          {"outputTensor2", { 5.0f, 6.0f, 7.0f, 8.0f } } } );
}

BOOST_FIXTURE_TEST_CASE() [4/8]

BOOST_FIXTURE_TEST_CASE	(	SimpleSplit3DFloat32	,
		SimpleSplit3DFixtureFloat32
	)

Definition at line 162 of file Split.cpp.

{
    RunTest<3, armnn::DataType::Float32>(
        0,
        { {"inputTensor", { 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f,
                            9.0f, 10.0f, 11.0f, 12.0f, 13.0f, 14.0f, 15.0f, 16.0f } } },
        { {"outputTensor1", { 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f } },
          {"outputTensor2", { 9.0f, 10.0f, 11.0f, 12.0f, 13.0f, 14.0f, 15.0f, 16.0f } } } );
}

BOOST_FIXTURE_TEST_CASE() [5/8]

BOOST_FIXTURE_TEST_CASE	(	ParseAxisOneSplitTwoUint8	,
		SimpleSplitFixtureUint8
	)

Definition at line 179 of file Split.cpp.

{

    RunTest<4, armnn::DataType::QAsymmU8>(
        0,
        { {"inputTensor", { 1, 2, 3, 4, 5, 6, 7, 8,
                            9, 10, 11, 12, 13, 14, 15, 16 } } },
        { {"outputTensor1", { 1, 2, 3, 4, 9, 10, 11, 12 } },
          {"outputTensor2", { 5, 6, 7, 8, 13, 14, 15, 16 } } });
}

BOOST_FIXTURE_TEST_CASE() [6/8]

BOOST_FIXTURE_TEST_CASE	(	ParseAxisThreeSplitTwoUint8	,
		SimpleSplitAxisThreeFixtureUint8
	)

Definition at line 197 of file Split.cpp.

{
    RunTest<4, armnn::DataType::QAsymmU8>(
        0,
        { {"inputTensor", { 1, 2, 3, 4, 5, 6, 7, 8,
                            9, 10, 11, 12, 13, 14, 15, 16 } } },
        { {"outputTensor1", { 1, 3, 5, 7, 9, 11, 13, 15 } },
          {"outputTensor2", { 2, 4, 6, 8, 10, 12, 14, 16 } } } );
}

BOOST_FIXTURE_TEST_CASE() [7/8]

BOOST_FIXTURE_TEST_CASE	(	SimpleSplit2DUint8	,
		SimpleSplit2DFixtureUint8
	)

Definition at line 214 of file Split.cpp.

{
    RunTest<2, armnn::DataType::QAsymmU8>(
            0,
            { {"inputTensor", { 1, 2, 3, 4, 5, 6, 7, 8 } } },
            { {"outputTensor1", { 1, 2, 3, 4 } },
              {"outputTensor2", { 5, 6, 7, 8 } } } );
}

BOOST_FIXTURE_TEST_CASE() [8/8]

BOOST_FIXTURE_TEST_CASE	(	SimpleSplit3DUint8	,
		SimpleSplit3DFixtureUint8
	)

Definition at line 230 of file Split.cpp.

References BOOST_AUTO_TEST_SUITE_END().

{
    RunTest<3, armnn::DataType::QAsymmU8>(
        0,
        { {"inputTensor", { 1, 2, 3, 4, 5, 6, 7, 8,
                            9, 10, 11, 12, 13, 14, 15, 16 } } },
        { {"outputTensor1", { 1, 2, 3, 4, 5, 6, 7, 8 } },
          {"outputTensor2", { 9, 10, 11, 12, 13, 14, 15, 16 } } } );
}