LoadModel.cpp File Reference

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

Go to the source code of this file.

Typedefs
using	ModelPtr = TfLiteParserImpl::ModelPtr
using	SubgraphPtr = TfLiteParserImpl::SubgraphPtr
using	OperatorPtr = TfLiteParserImpl::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 = TfLiteParserImpl::ModelPtr

Definition at line 12 of file LoadModel.cpp.

OperatorPtr

using OperatorPtr = TfLiteParserImpl::OperatorPtr

Definition at line 14 of file LoadModel.cpp.

SubgraphPtr

using SubgraphPtr = TfLiteParserImpl::SubgraphPtr

Definition at line 13 of file LoadModel.cpp.

Function Documentation

BOOST_AUTO_TEST_CASE() [1/4]

BOOST_AUTO_TEST_CASE

(

LoadNullBinary

)

Definition at line 221 of file LoadModel.cpp.

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

BOOST_AUTO_TEST_CASE() [2/4]

BOOST_AUTO_TEST_CASE

(

LoadInvalidBinary

)

Definition at line 226 of file LoadModel.cpp.

{
    std::string testData = "invalid data";
    BOOST_CHECK_THROW(TfLiteParserImpl::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 233 of file LoadModel.cpp.

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

BOOST_AUTO_TEST_CASE() [4/4]

BOOST_AUTO_TEST_CASE

(

LoadNullPtrFile

)

Definition at line 238 of file LoadModel.cpp.

References BOOST_AUTO_TEST_SUITE_END().

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

BOOST_FIXTURE_TEST_CASE() [1/2]

BOOST_FIXTURE_TEST_CASE	(	LoadModelFromBinary	,
		LoadModelFixture
	)

Definition at line 186 of file LoadModel.cpp.

{
    TfLiteParserImpl::ModelPtr model = TfLiteParserImpl::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 200 of file LoadModel.cpp.

References armnnUtils::Filesystem::NamedTempFile().

{
    using namespace fs;
    fs::path fname = armnnUtils::Filesystem::NamedTempFile("Armnn-tfLite-LoadModelFromFile-TempFile.csv");
    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");

    TfLiteParserImpl::ModelPtr model = TfLiteParserImpl::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);
}