BatchToSpaceND.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 (BatchToSpaceNDTest1, BatchToSpaceNDFixtureTest1)
	BOOST_FIXTURE_TEST_CASE (ParseBatchToSpaceNDTest2, BatchToSpaceNDFixtureTest2)
	BOOST_FIXTURE_TEST_CASE (ParseBatchToSpaceNDTest3, BatchToSpaceNDFixtureTest3)

Function Documentation

BOOST_FIXTURE_TEST_CASE() [1/3]

BOOST_FIXTURE_TEST_CASE	(	BatchToSpaceNDTest1	,
		BatchToSpaceNDFixtureTest1
	)

Definition at line 108 of file BatchToSpaceND.cpp.

{
    RunTest<4, armnn::DataType::Float32>
        (0,
         {{ "inputTensor",  { // Batch 0, Height 0, Width (2) x Channel (1)
                              1.0f, 3.0f,
                              // Batch 0, Height 1, Width (2) x Channel (1)
                              9.0f, 11.0f,

                              // Batch 1, Height 0, Width (2) x Channel (1)
                              2.0f, 4.0f,
                              // Batch 1, Height 1, Width (2) x Channel (1)
                              10.0f, 12.0f,

                              // Batch 2, Height 0, Width (2) x Channel (1)
                              5.0f, 7.0f,
                              // Batch 2, Height 1, Width (2) x Channel (1)
                              13.0f, 15.0f,

                              // Batch 3, Height 0, Width (2) x Channel (3)
                              6.0f, 8.0f,
                              // Batch 3, Height 1, Width (2) x Channel (1)
                              14.0f, 16.0f }}},
         {{ "outputTensor", { 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 }}});
}

BOOST_FIXTURE_TEST_CASE() [2/3]

BOOST_FIXTURE_TEST_CASE	(	ParseBatchToSpaceNDTest2	,
		BatchToSpaceNDFixtureTest2
	)

Definition at line 145 of file BatchToSpaceND.cpp.

{
    RunTest<4, armnn::DataType::Float32>
        (0,
         {{ "inputTensor",  { 1.0f, 2.0f, 3.0f, 4.0f }}},
         {{ "outputTensor", { // Batch 0, Height 0, Width (2) x Channel (1)
                              1.0f, 2.0f, 3.0f, 4.0f }}});
}

BOOST_FIXTURE_TEST_CASE() [3/3]

BOOST_FIXTURE_TEST_CASE	(	ParseBatchToSpaceNDTest3	,
		BatchToSpaceNDFixtureTest3
	)

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