LoadModel.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
using	SubgraphPtr = TfLiteParser::SubgraphPtr
using	OperatorPtr = TfLiteParser::OperatorPtr

Functions
	BOOST_FIXTURE_TEST_CASE (LoadModelFromBinary, LoadModelFixture)
	BOOST_FIXTURE_TEST_CASE (LoadModelFromFile, LoadModelFixture)
	BOOST_AUTO_TEST_CASE (LoadNullBinary)
	BOOST_AUTO_TEST_CASE (LoadInvalidBinary)
	BOOST_AUTO_TEST_CASE (LoadFileNotFound)
	BOOST_AUTO_TEST_CASE (LoadNullPtrFile)

Typedef Documentation

ModelPtr

using ModelPtr = TfLiteParser::ModelPtr

Definition at line 10 of file LoadModel.cpp.

OperatorPtr

using OperatorPtr = TfLiteParser::OperatorPtr

Definition at line 12 of file LoadModel.cpp.

SubgraphPtr

using SubgraphPtr = TfLiteParser::SubgraphPtr

Definition at line 11 of file LoadModel.cpp.

Function Documentation

BOOST_AUTO_TEST_CASE() [1/4]

BOOST_AUTO_TEST_CASE

(

LoadNullBinary

)

Definition at line 218 of file LoadModel.cpp.

{
    BOOST_CHECK_THROW(TfLiteParser::LoadModelFromBinary(nullptr, 0), armnn::InvalidArgumentException);
}

BOOST_AUTO_TEST_CASE() [2/4]

BOOST_AUTO_TEST_CASE

(

LoadInvalidBinary

)

Definition at line 223 of file LoadModel.cpp.

{
    std::string testData = "invalid data";
    BOOST_CHECK_THROW(TfLiteParser::LoadModelFromBinary(reinterpret_cast<const uint8_t*>(&testData),
                                                        testData.length()), armnn::ParseException);
}

BOOST_AUTO_TEST_CASE() [3/4]

BOOST_AUTO_TEST_CASE

(

LoadFileNotFound

)

Definition at line 230 of file LoadModel.cpp.

{
    BOOST_CHECK_THROW(TfLiteParser::LoadModelFromFile("invalidfile.tflite"), armnn::FileNotFoundException);
}

BOOST_AUTO_TEST_CASE() [4/4]

BOOST_AUTO_TEST_CASE

(

LoadNullPtrFile

)

Definition at line 235 of file LoadModel.cpp.

References BOOST_AUTO_TEST_SUITE_END().

{
    BOOST_CHECK_THROW(TfLiteParser::LoadModelFromFile(nullptr), armnn::InvalidArgumentException);
}

BOOST_FIXTURE_TEST_CASE() [1/2]

BOOST_FIXTURE_TEST_CASE	(	LoadModelFromBinary	,
		LoadModelFixture
	)

Definition at line 184 of file LoadModel.cpp.

{
    TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromBinary(m_GraphBinary.data(), m_GraphBinary.size());
    CheckModel(model, 3, 2, { tflite::BuiltinOperator_AVERAGE_POOL_2D, tflite::BuiltinOperator_CONV_2D },
               2, "Test loading a model", 2);
    CheckSubgraph(model->subgraphs[0], 2, { 1 }, { 0 }, 1, "");
    CheckSubgraph(model->subgraphs[1], 3, { 0 }, { 1 }, 1, "");
    CheckOperator(model->subgraphs[0]->operators[0], 0, { 1 }, { 0 }, tflite::BuiltinOptions_Pool2DOptions,
                  tflite::CustomOptionsFormat_FLEXBUFFERS);
    CheckOperator(model->subgraphs[1]->operators[0], 1, { 0, 2 }, { 1 }, tflite::BuiltinOptions_Conv2DOptions,
                  tflite::CustomOptionsFormat_FLEXBUFFERS);
}

BOOST_FIXTURE_TEST_CASE() [2/2]

BOOST_FIXTURE_TEST_CASE	(	LoadModelFromFile	,
		LoadModelFixture
	)

Definition at line 197 of file LoadModel.cpp.

{
    using namespace boost::filesystem;
    std::string fname = unique_path(temp_directory_path() / "%%%%-%%%%-%%%%.tflite").string();
    bool saved = flatbuffers::SaveFile(fname.c_str(),
                                       reinterpret_cast<char *>(m_GraphBinary.data()),
                                       m_GraphBinary.size(), true);
    BOOST_CHECK_MESSAGE(saved, "Cannot save test file");

    TfLiteParser::ModelPtr model = TfLiteParser::LoadModelFromFile(fname.c_str());
    CheckModel(model, 3, 2, { tflite::BuiltinOperator_AVERAGE_POOL_2D, tflite::BuiltinOperator_CONV_2D },
               2, "Test loading a model", 2);
    CheckSubgraph(model->subgraphs[0], 2, { 1 }, { 0 }, 1, "");
    CheckSubgraph(model->subgraphs[1], 3, { 0 }, { 1 }, 1, "");
    CheckOperator(model->subgraphs[0]->operators[0], 0, { 1 }, { 0 }, tflite::BuiltinOptions_Pool2DOptions,
                  tflite::CustomOptionsFormat_FLEXBUFFERS);
    CheckOperator(model->subgraphs[1]->operators[0], 1, { 0, 2 }, { 1 }, tflite::BuiltinOptions_Conv2DOptions,
                  tflite::CustomOptionsFormat_FLEXBUFFERS);
    remove(fname);
}