Shape.cpp File Reference

#include <boost/test/unit_test.hpp>
#include "armnnTfParser/ITfParser.hpp"
#include "ParserPrototxtFixture.hpp"

Go to the source code of this file.

Functions
	BOOST_FIXTURE_TEST_CASE (ParseShape, ShapeFixture)

Function Documentation

BOOST_FIXTURE_TEST_CASE()

BOOST_FIXTURE_TEST_CASE	(	ParseShape	,
		ShapeFixture
	)

Definition at line 86 of file Shape.cpp.

References BOOST_AUTO_TEST_SUITE_END().

{
    // Note: the test's output cannot be an int32 const layer, because ARMNN only supports u8 and float layers.
    //       For that reason I added a reshape layer which reshapes the input to its original dimensions.
    RunTest<2>({ 0.0f, 1.0f, 2.0f, 3.0f }, { 0.0f, 1.0f, 2.0f, 3.0f });
}