Network.hpp File Reference

#include <armnn/DescriptorsFwd.hpp>
#include <armnn/LstmParams.hpp>
#include <armnn/QuantizedLstmParams.hpp>
#include <armnn/TensorFwd.hpp>
#include <armnn/Types.hpp>
#include <armnn/INetwork.hpp>
#include <string>
#include <vector>
#include <map>
#include <memory>
#include "Graph.hpp"
#include "Layer.hpp"
#include "OptimizedNetworkImpl.hpp"

Go to the source code of this file.

Classes
class	NetworkImpl
	Private implementation of INetwork. More...
struct	OptimizationResult

Namespaces
	armnn
	Copyright (c) 2021 ARM Limited and Contributors.

Typedefs
using	NetworkImplPtr = std::unique_ptr< NetworkImpl, void(*)(NetworkImpl *network)>
using	BackendsMap = std::map< BackendId, std::unique_ptr< class IBackendInternal > >

Functions
BackendsMap	CreateSupportedBackends (TensorHandleFactoryRegistry &handleFactoryRegistry, BackendSettings &backendSettings)
OptimizationResult	SelectTensorHandleStrategy (Graph &optGraph, BackendsMap &backends, TensorHandleFactoryRegistry &registry, bool importEnabled, Optional< std::vector< std::string > &> errMessages)
OptimizationResult	AssignBackends (OptimizedNetworkImpl *optNetObjPtr, BackendSettings &backendSettings, Graph::Iterator &firstLayer, Graph::Iterator &lastLayer, Optional< std::vector< std::string > &> errMessages)