SpaceToBatchND.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 (SpaceToBatchNdSimpleTest, SpaceToBatchNDFixtureSimpleTest)
	BOOST_FIXTURE_TEST_CASE (SpaceToBatchNdMultipleInputBatchesTest, SpaceToBatchNDFixtureMultipleInputBatchesTest)
	BOOST_FIXTURE_TEST_CASE (SpaceToBatchNdPaddingTest, SpaceToBatchNDFixturePaddingTest)

Function Documentation

BOOST_FIXTURE_TEST_CASE() [1/3]

BOOST_FIXTURE_TEST_CASE	(	SpaceToBatchNdSimpleTest	,
		SpaceToBatchNDFixtureSimpleTest
	)

Definition at line 108 of file SpaceToBatchND.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 }}},
         {{ "outputTensor", { 1.0f, 3.0f,  9.0f, 11.0f,
                              2.0f, 4.0f, 10.0f, 12.0f,
                              5.0f, 7.0f, 13.0f, 15.0f,
                              6.0f, 8.0f, 14.0f, 16.0f }}});
}

BOOST_FIXTURE_TEST_CASE() [2/3]

BOOST_FIXTURE_TEST_CASE	(	SpaceToBatchNdMultipleInputBatchesTest	,
		SpaceToBatchNDFixtureMultipleInputBatchesTest
	)

Definition at line 131 of file SpaceToBatchND.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 }}},
         {{ "outputTensor", {  1.0f, 3.0f,  9.0f, 11.0f,
                               2.0f, 4.0f, 10.0f, 12.0f,
                               5.0f, 7.0f, 13.0f, 15.0f,
                               6.0f, 8.0f, 14.0f, 16.0f }}});
}

BOOST_FIXTURE_TEST_CASE() [3/3]

BOOST_FIXTURE_TEST_CASE	(	SpaceToBatchNdPaddingTest	,
		SpaceToBatchNDFixturePaddingTest
	)

Definition at line 153 of file SpaceToBatchND.cpp.

References BOOST_AUTO_TEST_SUITE_END().

{
    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 }}},
         {{ "outputTensor", {  0.0f, 0.0f,
                               0.0f, 5.0f,

                               0.0f, 0.0f,
                               0.0f, 6.0f,

                               0.0f, 1.0f,
                               0.0f, 7.0f,

                               0.0f, 2.0f,
                               0.0f, 8.0f,

                               0.0f, 3.0f,
                               0.0f, 9.0f,

                               0.0f, 4.0f,
                               0.0f, 10.0f, }}});
}