Squeeze.cpp File Reference

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

Go to the source code of this file.

Typedefs
typedef SqueezeFixture< false, false >	ImpliedDimensionsSqueezeFixture
typedef SqueezeFixture< true, false >	ExplicitDimensionZeroSqueezeFixture
typedef SqueezeFixture< false, true >	ExplicitDimensionOneSqueezeFixture
typedef SqueezeFixture< true, true >	ExplicitDimensionsSqueezeFixture

Functions
	BOOST_FIXTURE_TEST_CASE (ParseImplicitSqueeze, ImpliedDimensionsSqueezeFixture)
	BOOST_FIXTURE_TEST_CASE (ParseDimensionZeroSqueeze, ExplicitDimensionZeroSqueezeFixture)
	BOOST_FIXTURE_TEST_CASE (ParseDimensionOneSqueeze, ExplicitDimensionOneSqueezeFixture)
	BOOST_FIXTURE_TEST_CASE (ParseExplicitDimensionsSqueeze, ExplicitDimensionsSqueezeFixture)

Typedef Documentation

ExplicitDimensionOneSqueezeFixture

typedef SqueezeFixture<false, true> ExplicitDimensionOneSqueezeFixture

Definition at line 72 of file Squeeze.cpp.

ExplicitDimensionsSqueezeFixture

typedef SqueezeFixture<true, true> ExplicitDimensionsSqueezeFixture

Definition at line 73 of file Squeeze.cpp.

ExplicitDimensionZeroSqueezeFixture

typedef SqueezeFixture<true, false> ExplicitDimensionZeroSqueezeFixture

Definition at line 71 of file Squeeze.cpp.

ImpliedDimensionsSqueezeFixture

typedef SqueezeFixture<false, false> ImpliedDimensionsSqueezeFixture

Definition at line 70 of file Squeeze.cpp.

Function Documentation

BOOST_FIXTURE_TEST_CASE() [1/4]

BOOST_FIXTURE_TEST_CASE	(	ParseImplicitSqueeze	,
		ImpliedDimensionsSqueezeFixture
	)

Definition at line 75 of file Squeeze.cpp.

{
    BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo("Squeeze").second.GetShape() ==
               armnn::TensorShape({2,2})));
    RunTest<2>({ 1.0f, 2.0f, 3.0f, 4.0f },
               { 1.0f, 2.0f, 3.0f, 4.0f });
}

BOOST_FIXTURE_TEST_CASE() [2/4]

BOOST_FIXTURE_TEST_CASE	(	ParseDimensionZeroSqueeze	,
		ExplicitDimensionZeroSqueezeFixture
	)

Definition at line 83 of file Squeeze.cpp.

{
    BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo("Squeeze").second.GetShape() ==
               armnn::TensorShape({1,2,2})));
    RunTest<3>({ 1.0f, 2.0f, 3.0f, 4.0f },
               { 1.0f, 2.0f, 3.0f, 4.0f });
}

BOOST_FIXTURE_TEST_CASE() [3/4]

BOOST_FIXTURE_TEST_CASE	(	ParseDimensionOneSqueeze	,
		ExplicitDimensionOneSqueezeFixture
	)

Definition at line 91 of file Squeeze.cpp.

{
    BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo("Squeeze").second.GetShape() ==
               armnn::TensorShape({1,2,2})));
    RunTest<3>({ 1.0f, 2.0f, 3.0f, 4.0f },
               { 1.0f, 2.0f, 3.0f, 4.0f });
}

BOOST_FIXTURE_TEST_CASE() [4/4]

BOOST_FIXTURE_TEST_CASE	(	ParseExplicitDimensionsSqueeze	,
		ExplicitDimensionsSqueezeFixture
	)

Definition at line 99 of file Squeeze.cpp.

References BOOST_AUTO_TEST_SUITE_END().

{
    BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo("Squeeze").second.GetShape() ==
               armnn::TensorShape({2,2})));
    RunTest<2>({ 1.0f, 2.0f, 3.0f, 4.0f },
               { 1.0f, 2.0f, 3.0f, 4.0f });
}