CreateNetworkImpl< IParser > Struct Template Reference

#include <InferenceModel.hpp>

Public Types
using	Params = InferenceModelInternal::Params

Static Public Member Functions
static armnn::INetworkPtr	Create (const Params &params, std::vector< armnn::BindingPointInfo > &inputBindings, std::vector< armnn::BindingPointInfo > &outputBindings)

Detailed Description

template<typename IParser>
struct CreateNetworkImpl< IParser >

Definition at line 127 of file InferenceModel.hpp.

Member Typedef Documentation

Params

using Params = InferenceModelInternal::Params

Definition at line 130 of file InferenceModel.hpp.

Member Function Documentation

Create()

static armnn::INetworkPtr Create ( const Params &  params, std::vector< armnn::BindingPointInfo > &  inputBindings, std::vector< armnn::BindingPointInfo > &  outputBindings  )

inlinestatic

Definition at line 132 of file InferenceModel.hpp.

References ARMNN_ASSERT, ARMNN_SCOPED_HEAP_PROFILING, CHECK_LOCATION, BindingPointInfo::m_BindingId, Params::m_InferOutputShape, Params::m_InputBindings, Params::m_InputShapes, Params::m_IsModelBinary, Params::m_ModelPath, Params::m_OutputBindings, Params::m_ParseUnsupported, Params::m_SubgraphId, BindingPointInfo::m_TensorInfo, and armnn::numeric_cast().

Referenced by InferenceModel< IParser, TDataType >::InferenceModel().

    {
        const std::string& modelPath = params.m_ModelPath;

        // Create a network from a file on disk
        auto parser(IParser::Create());

        std::map<std::string, armnn::TensorShape> inputShapes;
        if (!params.m_InputShapes.empty())
        {
            const size_t numInputShapes   = params.m_InputShapes.size();
            const size_t numInputBindings = params.m_InputBindings.size();
            if (numInputShapes < numInputBindings)
            {
                throw armnn::Exception(fmt::format(
                    "Not every input has its tensor shape specified: expected={0}, got={1}",
                    numInputBindings, numInputShapes));
            }

            for (size_t i = 0; i < numInputShapes; i++)
            {
                inputShapes[params.m_InputBindings[i]] = params.m_InputShapes[i];
            }
        }

        std::vector<std::string> requestedOutputs = params.m_OutputBindings;
        armnn::INetworkPtr network{nullptr, [](armnn::INetwork *){}};

        {
            ARMNN_SCOPED_HEAP_PROFILING("Parsing");
            // Handle text and binary input differently by calling the corresponding parser function
            network = (params.m_IsModelBinary ?
                parser->CreateNetworkFromBinaryFile(modelPath.c_str(), inputShapes, requestedOutputs) :
                parser->CreateNetworkFromTextFile(modelPath.c_str(), inputShapes, requestedOutputs));
        }

        for (const std::string& inputLayerName : params.m_InputBindings)
        {
            inputBindings.push_back(parser->GetNetworkInputBindingInfo(inputLayerName));
        }

        for (const std::string& outputLayerName : params.m_OutputBindings)
        {
            outputBindings.push_back(parser->GetNetworkOutputBindingInfo(outputLayerName));
        }

        return network;
    }

---

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

• tests/InferenceModel.hpp