#include <boost/test/unit_test.hpp>
#include "ParserFlatbuffersFixture.hpp"
#include "../TfLiteParser.hpp"
#include <string>
#include <iostream>

Functions
	BOOST_FIXTURE_TEST_CASE (ParseSqueezeWithSqueezeDims, SqueezeFixtureWithSqueezeDims)

	BOOST_FIXTURE_TEST_CASE (ParseSqueezeWithoutSqueezeDims, SqueezeFixtureWithoutSqueezeDims)

	BOOST_FIXTURE_TEST_CASE (ParseSqueezeInvalidInput, SqueezeFixtureWithInvalidInput)

	BOOST_FIXTURE_TEST_CASE (ParseSqueezeInvalidSqueezeDims, SqueezeFixtureWithSqueezeDimsSizeInvalid)

	BOOST_FIXTURE_TEST_CASE (ParseSqueezeNegativeSqueezeDims, SqueezeFixtureWithNegativeSqueezeDims)

Function Documentation

◆ BOOST_FIXTURE_TEST_CASE() [1/5]

BOOST_FIXTURE_TEST_CASE	(	ParseSqueezeWithSqueezeDims	,
		SqueezeFixtureWithSqueezeDims
	)

Definition at line 85 of file Squeeze.cpp.

 {
     SetupSingleInputSingleOutput("inputTensor", "outputTensor");
     RunTest<3, armnn::DataType::QAsymmU8>(0, { 1, 2, 3, 4 }, { 1, 2, 3, 4 });
     BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo(0, "outputTensor").second.GetShape()
         == armnn::TensorShape({2,2,1})));
 
 }

◆ BOOST_FIXTURE_TEST_CASE() [2/5]

BOOST_FIXTURE_TEST_CASE	(	ParseSqueezeWithoutSqueezeDims	,
		SqueezeFixtureWithoutSqueezeDims
	)

Definition at line 99 of file Squeeze.cpp.

 {
     SetupSingleInputSingleOutput("inputTensor", "outputTensor");
     RunTest<2, armnn::DataType::QAsymmU8>(0, { 1, 2, 3, 4 }, { 1, 2, 3, 4 });
     BOOST_TEST((m_Parser->GetNetworkOutputBindingInfo(0, "outputTensor").second.GetShape()
         == armnn::TensorShape({2,2})));
 }

◆ BOOST_FIXTURE_TEST_CASE() [3/5]

BOOST_FIXTURE_TEST_CASE	(	ParseSqueezeInvalidInput	,
		SqueezeFixtureWithInvalidInput
	)

Definition at line 112 of file Squeeze.cpp.

References armnn::MaxNumOfTensorDimensions.

 {
     static_assert(armnn::MaxNumOfTensorDimensions == 5, "Please update SqueezeFixtureWithInvalidInput");
     BOOST_CHECK_THROW((SetupSingleInputSingleOutput("inputTensor", "outputTensor")),
                       armnn::InvalidArgumentException);
 }

◆ BOOST_FIXTURE_TEST_CASE() [4/5]

BOOST_FIXTURE_TEST_CASE	(	ParseSqueezeInvalidSqueezeDims	,
		SqueezeFixtureWithSqueezeDimsSizeInvalid
	)

Definition at line 126 of file Squeeze.cpp.

 {
     BOOST_CHECK_THROW((SetupSingleInputSingleOutput("inputTensor", "outputTensor")), armnn::ParseException);
 }

◆ BOOST_FIXTURE_TEST_CASE() [5/5]

BOOST_FIXTURE_TEST_CASE	(	ParseSqueezeNegativeSqueezeDims	,
		SqueezeFixtureWithNegativeSqueezeDims
	)

Definition at line 139 of file Squeeze.cpp.

References BOOST_AUTO_TEST_SUITE_END().