ParserFlatbuffersSerializeFixture Struct Reference

#include <ParserFlatbuffersSerializeFixture.hpp>

Inheritance diagram for ParserFlatbuffersSerializeFixture:

Public Member Functions
	ParserFlatbuffersSerializeFixture ()
void	Setup ()
void	SetupSingleInputSingleOutput (const std::string &inputName, const std::string &outputName)
bool	ReadStringToBinary ()
template<std::size_t NumOutputDimensions, armnn::DataType ArmnnType, typename DataType = armnn::ResolveType<ArmnnType>>
void	RunTest (unsigned int layersId, const std::vector< DataType > &inputData, const std::vector< DataType > &expectedOutputData)
template<std::size_t NumOutputDimensions, armnn::DataType ArmnnInputType, armnn::DataType ArmnnOutputType, typename InputDataType = armnn::ResolveType<ArmnnInputType>, typename OutputDataType = armnn::ResolveType<ArmnnOutputType>>
void	RunTest (unsigned int layersId, const std::vector< InputDataType > &inputData, const std::vector< OutputDataType > &expectedOutputData)
template<std::size_t NumOutputDimensions, armnn::DataType ArmnnType, typename DataType = armnn::ResolveType<ArmnnType>>
void	RunTest (unsigned int layersId, const std::map< std::string, std::vector< DataType >> &inputData, const std::map< std::string, std::vector< DataType >> &expectedOutputData)
template<std::size_t NumOutputDimensions, armnn::DataType ArmnnInputType, armnn::DataType ArmnnOutputType, typename InputDataType = armnn::ResolveType<ArmnnInputType>, typename OutputDataType = armnn::ResolveType<ArmnnOutputType>>
void	RunTest (unsigned int layersId, const std::map< std::string, std::vector< InputDataType >> &inputData, const std::map< std::string, std::vector< OutputDataType >> &expectedOutputData)
void	CheckTensors (const TensorRawPtr &tensors, size_t shapeSize, const std::vector< int32_t > &shape, armnnSerializer::TensorInfo tensorType, const std::string &name, const float scale, const int64_t zeroPoint)

Public Attributes
std::vector< uint8_t >	m_GraphBinary
std::string	m_JsonString
std::unique_ptr< IDeserializer, void(*)(IDeserializer *parser)>	m_Parser
armnn::IRuntimePtr	m_Runtime
armnn::NetworkId	m_NetworkIdentifier
std::string	m_SingleInputName
std::string	m_SingleOutputName

Detailed Description

Definition at line 28 of file ParserFlatbuffersSerializeFixture.hpp.

Constructor & Destructor Documentation

ParserFlatbuffersSerializeFixture()

ParserFlatbuffersSerializeFixture ( )

inline

Definition at line 30 of file ParserFlatbuffersSerializeFixture.hpp.

                                        :
        m_Parser(IDeserializer::Create()),
        m_Runtime(armnn::IRuntime::Create(armnn::IRuntime::CreationOptions())),
        m_NetworkIdentifier(-1)
    {
    }

Member Function Documentation

CheckTensors()

void CheckTensors ( const TensorRawPtr &  tensors, size_t  shapeSize, const std::vector< int32_t > &  shape, armnnSerializer::TensorInfo  tensorType, const std::string &  name, const float  scale, const int64_t  zeroPoint  )

inline

Definition at line 154 of file ParserFlatbuffersSerializeFixture.hpp.

    {
        boost::ignore_unused(name);
        BOOST_CHECK_EQUAL(shapeSize, tensors->dimensions()->size());
        BOOST_CHECK_EQUAL_COLLECTIONS(shape.begin(), shape.end(),
                                      tensors->dimensions()->begin(), tensors->dimensions()->end());
        BOOST_CHECK_EQUAL(tensorType.dataType(), tensors->dataType());
        BOOST_CHECK_EQUAL(scale, tensors->quantizationScale());
        BOOST_CHECK_EQUAL(zeroPoint, tensors->quantizationOffset());
    }

ReadStringToBinary()

bool ReadStringToBinary ( )

inline

Definition at line 92 of file ParserFlatbuffersSerializeFixture.hpp.

References deserialize_schema_end, deserialize_schema_start, and RunTest().

Referenced by Setup().

    {
        std::string schemafile(&deserialize_schema_start, &deserialize_schema_end);

        // parse schema first, so we can use it to parse the data after
        flatbuffers::Parser parser;

        bool ok = parser.Parse(schemafile.c_str());
        BOOST_ASSERT_MSG(ok, "Failed to parse schema file");

        ok &= parser.Parse(m_JsonString.c_str());
        BOOST_ASSERT_MSG(ok, "Failed to parse json input");

        if (!ok)
        {
            return false;
        }

        {
            const uint8_t* bufferPtr = parser.builder_.GetBufferPointer();
            size_t size = static_cast<size_t>(parser.builder_.GetSize());
            m_GraphBinary.assign(bufferPtr, bufferPtr+size);
        }
        return ok;
    }

RunTest() [1/4]

void RunTest	(	unsigned int	layersId,
		const std::vector< DataType > &	inputData,
		const std::vector< DataType > &	expectedOutputData
	)

Executes the network with the given input tensor and checks the result against the given output tensor. This overload assumes the network has a single input and a single output.

Definition at line 169 of file ParserFlatbuffersSerializeFixture.hpp.

Referenced by ReadStringToBinary().

{
    RunTest<NumOutputDimensions, ArmnnType, ArmnnType, DataType, DataType>(layersId, inputData, expectedOutputData);
}

RunTest() [2/4]

void RunTest	(	unsigned int	layersId,
		const std::vector< InputDataType > &	inputData,
		const std::vector< OutputDataType > &	expectedOutputData
	)

Definition at line 181 of file ParserFlatbuffersSerializeFixture.hpp.

References m_SingleInputName, and m_SingleOutputName.

{
    RunTest<NumOutputDimensions, ArmnnInputType, ArmnnOutputType>(layersId,
                                                                  { { m_SingleInputName, inputData } },
                                                                  { { m_SingleOutputName, expectedOutputData } });
}

RunTest() [3/4]

void RunTest	(	unsigned int	layersId,
		const std::map< std::string, std::vector< DataType >> &	inputData,
		const std::map< std::string, std::vector< DataType >> &	expectedOutputData
	)

Executes the network with the given input tensors and checks the results against the given output tensors. This overload supports multiple inputs and multiple outputs, identified by name.

Definition at line 191 of file ParserFlatbuffersSerializeFixture.hpp.

{
    RunTest<NumOutputDimensions, ArmnnType, ArmnnType, DataType, DataType>(layersId, inputData, expectedOutputData);
}

RunTest() [4/4]

void RunTest	(	unsigned int	layersId,
		const std::map< std::string, std::vector< InputDataType >> &	inputData,
		const std::map< std::string, std::vector< OutputDataType >> &	expectedOutputData
	)

Definition at line 203 of file ParserFlatbuffersSerializeFixture.hpp.

References CompareTensors(), m_NetworkIdentifier, m_Parser, m_Runtime, and armnn::VerifyTensorInfoDataType().

{
    auto ConvertBindingInfo = [](const armnnDeserializer::BindingPointInfo& bindingInfo)
        {
            return std::make_pair(bindingInfo.m_BindingId, bindingInfo.m_TensorInfo);
        };

    // Setup the armnn input tensors from the given vectors.
    armnn::InputTensors inputTensors;
    for (auto&& it : inputData)
    {
        armnn::BindingPointInfo bindingInfo = ConvertBindingInfo(
            m_Parser->GetNetworkInputBindingInfo(layersId, it.first));
        armnn::VerifyTensorInfoDataType(bindingInfo.second, ArmnnInputType);
        inputTensors.push_back({ bindingInfo.first, armnn::ConstTensor(bindingInfo.second, it.second.data()) });
    }

    // Allocate storage for the output tensors to be written to and setup the armnn output tensors.
    std::map<std::string, boost::multi_array<OutputDataType, NumOutputDimensions>> outputStorage;
    armnn::OutputTensors outputTensors;
    for (auto&& it : expectedOutputData)
    {
        armnn::BindingPointInfo bindingInfo = ConvertBindingInfo(
            m_Parser->GetNetworkOutputBindingInfo(layersId, it.first));
        armnn::VerifyTensorInfoDataType(bindingInfo.second, ArmnnOutputType);
        outputStorage.emplace(it.first, MakeTensor<OutputDataType, NumOutputDimensions>(bindingInfo.second));
        outputTensors.push_back(
                { bindingInfo.first, armnn::Tensor(bindingInfo.second, outputStorage.at(it.first).data()) });
    }

    m_Runtime->EnqueueWorkload(m_NetworkIdentifier, inputTensors, outputTensors);

    // Compare each output tensor to the expected values
    for (auto&& it : expectedOutputData)
    {
        armnn::BindingPointInfo bindingInfo = ConvertBindingInfo(
            m_Parser->GetNetworkOutputBindingInfo(layersId, it.first));
        auto outputExpected = MakeTensor<OutputDataType, NumOutputDimensions>(bindingInfo.second, it.second);
        BOOST_TEST(CompareTensors(outputExpected, outputStorage[it.first]));
    }
}

Setup()

void Setup ( )

inline

Definition at line 48 of file ParserFlatbuffersSerializeFixture.hpp.

References armnn::CpuRef, armnn::Optimize(), ReadStringToBinary(), and armnn::Success.

Referenced by SetupSingleInputSingleOutput().

    {
        bool ok = ReadStringToBinary();
        if (!ok)
        {
            throw armnn::Exception("LoadNetwork failed while reading binary input");
        }

        armnn::INetworkPtr network =
                m_Parser->CreateNetworkFromBinary(m_GraphBinary);

        if (!network)
        {
            throw armnn::Exception("The parser failed to create an ArmNN network");
        }

        auto optimized = Optimize(*network, {armnn::Compute::CpuRef},
                                  m_Runtime->GetDeviceSpec());

        std::string errorMessage;
        armnn::Status ret = m_Runtime->LoadNetwork(m_NetworkIdentifier, move(optimized), errorMessage);

        if (ret != armnn::Status::Success)
        {
            throw armnn::Exception(
                    boost::str(
                            boost::format("The runtime failed to load the network. "
                                          "Error was: %1%. in %2% [%3%:%4%]") %
                            errorMessage %
                            __func__ %
                            __FILE__ %
                            __LINE__));
        }

    }

SetupSingleInputSingleOutput()

void SetupSingleInputSingleOutput ( const std::string &  inputName, const std::string &  outputName  )

inline

Definition at line 84 of file ParserFlatbuffersSerializeFixture.hpp.

References Setup().

    {
        // Store the input and output name so they don't need to be passed to the single-input-single-output RunTest().
        m_SingleInputName = inputName;
        m_SingleOutputName = outputName;
        Setup();
    }

Member Data Documentation

m_GraphBinary

std::vector<uint8_t> m_GraphBinary

Definition at line 37 of file ParserFlatbuffersSerializeFixture.hpp.

m_JsonString

std::string m_JsonString

Definition at line 38 of file ParserFlatbuffersSerializeFixture.hpp.

m_NetworkIdentifier

armnn::NetworkId m_NetworkIdentifier

Definition at line 41 of file ParserFlatbuffersSerializeFixture.hpp.

Referenced by RunTest().

m_Parser

std::unique_ptr<IDeserializer, void (*)(IDeserializer* parser)> m_Parser

Definition at line 39 of file ParserFlatbuffersSerializeFixture.hpp.

Referenced by RunTest().

m_Runtime

armnn::IRuntimePtr m_Runtime

Definition at line 40 of file ParserFlatbuffersSerializeFixture.hpp.

Referenced by RunTest().

m_SingleInputName

std::string m_SingleInputName

If the single-input-single-output overload of Setup() is called, these will store the input and output name so they don't need to be passed to the single-input-single-output overload of RunTest().

Definition at line 45 of file ParserFlatbuffersSerializeFixture.hpp.

Referenced by RunTest().

m_SingleOutputName

std::string m_SingleOutputName

Definition at line 46 of file ParserFlatbuffersSerializeFixture.hpp.

Referenced by RunTest().

---

The documentation for this struct was generated from the following file:

• src/armnnDeserializer/test/ParserFlatbuffersSerializeFixture.hpp