GetTensorIds.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 = TfLiteParser::ModelPtr

Functions
	BOOST_FIXTURE_TEST_CASE (GetEmptyInputTensorIds, GetEmptyTensorIdsFixture)
	BOOST_FIXTURE_TEST_CASE (GetEmptyOutputTensorIds, GetEmptyTensorIdsFixture)
	BOOST_FIXTURE_TEST_CASE (GetInputTensorIds, GetInputOutputTensorIdsFixture)
	BOOST_FIXTURE_TEST_CASE (GetOutputTensorIds, GetInputOutputTensorIdsFixture)
	BOOST_FIXTURE_TEST_CASE (GetInputTensorIdsNullModel, GetInputOutputTensorIdsFixture)
	BOOST_FIXTURE_TEST_CASE (GetOutputTensorIdsNullModel, GetInputOutputTensorIdsFixture)
	BOOST_FIXTURE_TEST_CASE (GetInputTensorIdsInvalidSubgraph, GetInputOutputTensorIdsFixture)
	BOOST_FIXTURE_TEST_CASE (GetOutputTensorIdsInvalidSubgraph, GetInputOutputTensorIdsFixture)
	BOOST_FIXTURE_TEST_CASE (GetInputTensorIdsInvalidOperator, GetInputOutputTensorIdsFixture)
	BOOST_FIXTURE_TEST_CASE (GetOutputTensorIdsInvalidOperator, GetInputOutputTensorIdsFixture)

Typedef Documentation

ModelPtr

using ModelPtr = TfLiteParser::ModelPtr

Definition at line 10 of file GetTensorIds.cpp.

Function Documentation

BOOST_FIXTURE_TEST_CASE() [1/10]

BOOST_FIXTURE_TEST_CASE	(	GetEmptyInputTensorIds	,
		GetEmptyTensorIdsFixture
	)

Definition at line 92 of file GetTensorIds.cpp.

{
    TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size());
    std::vector<int32_t> expectedIds = { };
    std::vector<int32_t> inputTensorIds = TfLiteParser::GetInputTensorIds(model, 0, 0);
    BOOST_CHECK_EQUAL_COLLECTIONS(expectedIds.begin(), expectedIds.end(),
                                  inputTensorIds.begin(), inputTensorIds.end());
}

BOOST_FIXTURE_TEST_CASE() [2/10]

BOOST_FIXTURE_TEST_CASE	(	GetEmptyOutputTensorIds	,
		GetEmptyTensorIdsFixture
	)

Definition at line 101 of file GetTensorIds.cpp.

{
    TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size());
    std::vector<int32_t> expectedIds = { };
    std::vector<int32_t> outputTensorIds = TfLiteParser::GetOutputTensorIds(model, 0, 0);
    BOOST_CHECK_EQUAL_COLLECTIONS(expectedIds.begin(), expectedIds.end(),
                                  outputTensorIds.begin(), outputTensorIds.end());
}

BOOST_FIXTURE_TEST_CASE() [3/10]

BOOST_FIXTURE_TEST_CASE	(	GetInputTensorIds	,
		GetInputOutputTensorIdsFixture
	)

Definition at line 110 of file GetTensorIds.cpp.

{
    TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size());
    std::vector<int32_t> expectedInputIds = { 0, 1, 2 };
    std::vector<int32_t> inputTensorIds = TfLiteParser::GetInputTensorIds(model, 0, 0);
    BOOST_CHECK_EQUAL_COLLECTIONS(expectedInputIds.begin(), expectedInputIds.end(),
                                  inputTensorIds.begin(), inputTensorIds.end());
}

BOOST_FIXTURE_TEST_CASE() [4/10]

BOOST_FIXTURE_TEST_CASE	(	GetOutputTensorIds	,
		GetInputOutputTensorIdsFixture
	)

Definition at line 119 of file GetTensorIds.cpp.

{
    TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size());
    std::vector<int32_t> expectedOutputIds = { 3 };
    std::vector<int32_t> outputTensorIds = TfLiteParser::GetOutputTensorIds(model, 0, 0);
    BOOST_CHECK_EQUAL_COLLECTIONS(expectedOutputIds.begin(), expectedOutputIds.end(),
                                  outputTensorIds.begin(), outputTensorIds.end());
}

BOOST_FIXTURE_TEST_CASE() [5/10]

BOOST_FIXTURE_TEST_CASE	(	GetInputTensorIdsNullModel	,
		GetInputOutputTensorIdsFixture
	)

Definition at line 128 of file GetTensorIds.cpp.

{
    BOOST_CHECK_THROW(TfLiteParser::GetInputTensorIds(nullptr, 0, 0), armnn::ParseException);
}

BOOST_FIXTURE_TEST_CASE() [6/10]

BOOST_FIXTURE_TEST_CASE	(	GetOutputTensorIdsNullModel	,
		GetInputOutputTensorIdsFixture
	)

Definition at line 133 of file GetTensorIds.cpp.

{
    BOOST_CHECK_THROW(TfLiteParser::GetOutputTensorIds(nullptr, 0, 0), armnn::ParseException);
}

BOOST_FIXTURE_TEST_CASE() [7/10]

BOOST_FIXTURE_TEST_CASE	(	GetInputTensorIdsInvalidSubgraph	,
		GetInputOutputTensorIdsFixture
	)

Definition at line 138 of file GetTensorIds.cpp.

{
    TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size());
    BOOST_CHECK_THROW(TfLiteParser::GetInputTensorIds(model, 1, 0), armnn::ParseException);
}

BOOST_FIXTURE_TEST_CASE() [8/10]

BOOST_FIXTURE_TEST_CASE	(	GetOutputTensorIdsInvalidSubgraph	,
		GetInputOutputTensorIdsFixture
	)

Definition at line 144 of file GetTensorIds.cpp.

{
    TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size());
    BOOST_CHECK_THROW(TfLiteParser::GetOutputTensorIds(model, 1, 0), armnn::ParseException);
}

BOOST_FIXTURE_TEST_CASE() [9/10]

BOOST_FIXTURE_TEST_CASE	(	GetInputTensorIdsInvalidOperator	,
		GetInputOutputTensorIdsFixture
	)

Definition at line 150 of file GetTensorIds.cpp.

{
    TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size());
    BOOST_CHECK_THROW(TfLiteParser::GetInputTensorIds(model, 0, 1), armnn::ParseException);
}

BOOST_FIXTURE_TEST_CASE() [10/10]

BOOST_FIXTURE_TEST_CASE	(	GetOutputTensorIdsInvalidOperator	,
		GetInputOutputTensorIdsFixture
	)

Definition at line 156 of file GetTensorIds.cpp.

References BOOST_AUTO_TEST_SUITE_END().

{
    TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size());
    BOOST_CHECK_THROW(TfLiteParser::GetOutputTensorIds(model, 0, 1), armnn::ParseException);
}