GetSubgraphInputsOutputs.cpp File Reference

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

Go to the source code of this file.

Typedefs
using	ModelPtr = TfLiteParserImpl::ModelPtr
using	TensorRawPtr = TfLiteParserImpl::TensorRawPtr

Functions
	BOOST_FIXTURE_TEST_CASE (GetEmptySubgraphInputs, GetEmptySubgraphInputsOutputsFixture)
	BOOST_FIXTURE_TEST_CASE (GetEmptySubgraphOutputs, GetEmptySubgraphInputsOutputsFixture)
	BOOST_FIXTURE_TEST_CASE (GetSubgraphInputs, GetSubgraphInputsOutputsFixture)
	BOOST_FIXTURE_TEST_CASE (GetSubgraphOutputsSimpleQuantized, GetSubgraphInputsOutputsFixture)
	BOOST_FIXTURE_TEST_CASE (GetSubgraphInputsEmptyMinMax, GetSubgraphInputsOutputsFixture)
	BOOST_FIXTURE_TEST_CASE (GetSubgraphOutputs, GetSubgraphInputsOutputsFixture)
	BOOST_AUTO_TEST_CASE (GetSubgraphInputsNullModel)
	BOOST_AUTO_TEST_CASE (GetSubgraphOutputsNullModel)
	BOOST_FIXTURE_TEST_CASE (GetSubgraphInputsInvalidSubgraph, GetSubgraphInputsOutputsFixture)
	BOOST_FIXTURE_TEST_CASE (GetSubgraphOutputsInvalidSubgraph, GetSubgraphInputsOutputsFixture)

Typedef Documentation

ModelPtr

using ModelPtr = TfLiteParserImpl::ModelPtr

Definition at line 10 of file GetSubgraphInputsOutputs.cpp.

TensorRawPtr

using TensorRawPtr = TfLiteParserImpl::TensorRawPtr

Definition at line 11 of file GetSubgraphInputsOutputs.cpp.

Function Documentation

BOOST_AUTO_TEST_CASE() [1/2]

BOOST_AUTO_TEST_CASE

(

GetSubgraphInputsNullModel

)

Definition at line 214 of file GetSubgraphInputsOutputs.cpp.

{
    BOOST_CHECK_THROW(TfLiteParserImpl::GetSubgraphInputs(nullptr, 0), armnn::ParseException);
}

BOOST_AUTO_TEST_CASE() [2/2]

BOOST_AUTO_TEST_CASE

(

GetSubgraphOutputsNullModel

)

Definition at line 219 of file GetSubgraphInputsOutputs.cpp.

{
    BOOST_CHECK_THROW(TfLiteParserImpl::GetSubgraphOutputs(nullptr, 0), armnn::ParseException);
}

BOOST_FIXTURE_TEST_CASE() [1/8]

BOOST_FIXTURE_TEST_CASE	(	GetEmptySubgraphInputs	,
		GetEmptySubgraphInputsOutputsFixture
	)

Definition at line 154 of file GetSubgraphInputsOutputs.cpp.

{
    TfLiteParserImpl::ModelPtr model = TfLiteParserImpl::LoadModelFromBinary(m_GraphBinary.data(),
                                                                             m_GraphBinary.size());
    TfLiteParserImpl::TensorIdRawPtrVector subgraphTensors = TfLiteParserImpl::GetSubgraphInputs(model, 0);
    BOOST_CHECK_EQUAL(0, subgraphTensors.size());
}

BOOST_FIXTURE_TEST_CASE() [2/8]

BOOST_FIXTURE_TEST_CASE	(	GetEmptySubgraphOutputs	,
		GetEmptySubgraphInputsOutputsFixture
	)

Definition at line 162 of file GetSubgraphInputsOutputs.cpp.

{
    TfLiteParserImpl::ModelPtr model = TfLiteParserImpl::LoadModelFromBinary(m_GraphBinary.data(),
                                                                             m_GraphBinary.size());
    TfLiteParserImpl::TensorIdRawPtrVector subgraphTensors = TfLiteParserImpl::GetSubgraphOutputs(model, 0);
    BOOST_CHECK_EQUAL(0, subgraphTensors.size());
}

BOOST_FIXTURE_TEST_CASE() [3/8]

BOOST_FIXTURE_TEST_CASE	(	GetSubgraphInputs	,
		GetSubgraphInputsOutputsFixture
	)

Definition at line 170 of file GetSubgraphInputsOutputs.cpp.

{
    TfLiteParserImpl::ModelPtr model = TfLiteParserImpl::LoadModelFromBinary(m_GraphBinary.data(),
                                                                             m_GraphBinary.size());
    TfLiteParserImpl::TensorIdRawPtrVector subgraphTensors = TfLiteParserImpl::GetSubgraphInputs(model, 0);
    BOOST_CHECK_EQUAL(1, subgraphTensors.size());
    BOOST_CHECK_EQUAL(1, subgraphTensors[0].first);
    CheckTensors(subgraphTensors[0].second, 4, { 1, 2, 2, 1 }, tflite::TensorType::TensorType_UINT8, 1,
                      "InputTensor", { -1.2f }, { 25.5f }, { 0.25f }, { 10 });
}

BOOST_FIXTURE_TEST_CASE() [4/8]

BOOST_FIXTURE_TEST_CASE	(	GetSubgraphOutputsSimpleQuantized	,
		GetSubgraphInputsOutputsFixture
	)

Definition at line 181 of file GetSubgraphInputsOutputs.cpp.

{
    TfLiteParserImpl::ModelPtr model = TfLiteParserImpl::LoadModelFromBinary(m_GraphBinary.data(),
                                                                             m_GraphBinary.size());
    TfLiteParserImpl::TensorIdRawPtrVector subgraphTensors = TfLiteParserImpl::GetSubgraphOutputs(model, 0);
    BOOST_CHECK_EQUAL(1, subgraphTensors.size());
    BOOST_CHECK_EQUAL(0, subgraphTensors[0].first);
    CheckTensors(subgraphTensors[0].second, 4, { 1, 1, 1, 1 }, tflite::TensorType::TensorType_UINT8, 0,
                      "OutputTensor", { 0.0f }, { 255.0f }, { 1.0f }, { 0 });
}

BOOST_FIXTURE_TEST_CASE() [5/8]

BOOST_FIXTURE_TEST_CASE	(	GetSubgraphInputsEmptyMinMax	,
		GetSubgraphInputsOutputsFixture
	)

Definition at line 192 of file GetSubgraphInputsOutputs.cpp.

{
    TfLiteParserImpl::ModelPtr model = TfLiteParserImpl::LoadModelFromBinary(m_GraphBinary.data(),
                                                                             m_GraphBinary.size());
    TfLiteParserImpl::TensorIdRawPtrVector subgraphTensors = TfLiteParserImpl::GetSubgraphInputs(model, 1);
    BOOST_CHECK_EQUAL(1, subgraphTensors.size());
    BOOST_CHECK_EQUAL(0, subgraphTensors[0].first);
    CheckTensors(subgraphTensors[0].second, 4, { 1, 3, 3, 1 }, tflite::TensorType::TensorType_UINT8, 0,
                      "ConvInputTensor", { }, { }, { 1.0f }, { 0 });
}

BOOST_FIXTURE_TEST_CASE() [6/8]

BOOST_FIXTURE_TEST_CASE	(	GetSubgraphOutputs	,
		GetSubgraphInputsOutputsFixture
	)

Definition at line 203 of file GetSubgraphInputsOutputs.cpp.

{
    TfLiteParserImpl::ModelPtr model = TfLiteParserImpl::LoadModelFromBinary(m_GraphBinary.data(),
                                                                             m_GraphBinary.size());
    TfLiteParserImpl::TensorIdRawPtrVector subgraphTensors = TfLiteParserImpl::GetSubgraphOutputs(model, 1);
    BOOST_CHECK_EQUAL(1, subgraphTensors.size());
    BOOST_CHECK_EQUAL(1, subgraphTensors[0].first);
    CheckTensors(subgraphTensors[0].second, 4, { 1, 1, 1, 1 }, tflite::TensorType::TensorType_UINT8, 1,
                      "ConvOutputTensor", { 0.0f }, { 511.0f }, { 2.0f }, { 0 });
}

BOOST_FIXTURE_TEST_CASE() [7/8]

BOOST_FIXTURE_TEST_CASE	(	GetSubgraphInputsInvalidSubgraph	,
		GetSubgraphInputsOutputsFixture
	)

Definition at line 224 of file GetSubgraphInputsOutputs.cpp.

{
    TfLiteParserImpl::ModelPtr model = TfLiteParserImpl::LoadModelFromBinary(m_GraphBinary.data(),
                                                                             m_GraphBinary.size());
    BOOST_CHECK_THROW(TfLiteParserImpl::GetSubgraphInputs(model, 2), armnn::ParseException);
}

BOOST_FIXTURE_TEST_CASE() [8/8]

BOOST_FIXTURE_TEST_CASE	(	GetSubgraphOutputsInvalidSubgraph	,
		GetSubgraphInputsOutputsFixture
	)

Definition at line 231 of file GetSubgraphInputsOutputs.cpp.

References BOOST_AUTO_TEST_SUITE_END().

{
    TfLiteParserImpl::ModelPtr model = TfLiteParserImpl::LoadModelFromBinary(m_GraphBinary.data(),
                                                                             m_GraphBinary.size());
    BOOST_CHECK_THROW(TfLiteParserImpl::GetSubgraphOutputs(model, 2), armnn::ParseException);
}