INetwork Class Reference

Main network class which provides the interface for building up a neural network. More...

#include <INetwork.hpp>

Public Member Functions
Status	PrintGraph ()
IConnectableLayer *	AddInputLayer (LayerBindingId id, const char *name=nullptr)
	Adds an input layer to the network.
IConnectableLayer *	AddArgMinMaxLayer (const ArgMinMaxDescriptor &desc, const char *name=nullptr)
	Adds an ArgMinMax layer to the network.
IConnectableLayer *	AddCastLayer (const char *name=nullptr)
	Adds a cast layer to the network.
IConnectableLayer *	AddComparisonLayer (const ComparisonDescriptor &comparisonDescriptor, const char *name=nullptr)
	Add a Comparison layer to the network.
IConnectableLayer *	AddConcatLayer (const ConcatDescriptor &concatDescriptor, const char *name=nullptr)
	Adds a concatenation layer to the network.
IConnectableLayer *	AddConvolution2dLayer (const Convolution2dDescriptor &convolution2dDescriptor, const char *name=nullptr)
	Adds a 2D convolution layer to the network.
IConnectableLayer *	AddConvolution3dLayer (const Convolution3dDescriptor &convolution3dDescriptor, const char *name=nullptr)
	Adds a 3D convolution layer to the network.
IConnectableLayer *	AddDepthToSpaceLayer (const DepthToSpaceDescriptor &depthToSpaceDescriptor, const char *name=nullptr)
	Adds a depth to space layer to the network.
IConnectableLayer *	AddDepthwiseConvolution2dLayer (const DepthwiseConvolution2dDescriptor &convolution2dDescriptor, const char *name=nullptr)
	Adds a 2D depthwise convolution layer to the network.
IConnectableLayer *	AddDequantizeLayer (const char *name=nullptr)
	Adds a Dequantize layer to the network.
IConnectableLayer *	AddDetectionPostProcessLayer (const DetectionPostProcessDescriptor &descriptor, const ConstTensor &anchors, const char *name=nullptr)
	Adds a Detection PostProcess layer to the network.
IConnectableLayer *	AddElementwiseBinaryLayer (const ElementwiseBinaryDescriptor &elementwiseBinaryDescriptor, const char *name=nullptr)
	Add an ElementwiseBinary layer to the network.
IConnectableLayer *	AddElementwiseUnaryLayer (const ElementwiseUnaryDescriptor &elementwiseUnaryDescriptor, const char *name=nullptr)
	Add an ElementwiseUnary layer to the network.
IConnectableLayer *	AddFillLayer (const FillDescriptor &fillDescriptor, const char *name=nullptr)
	Add an Fill layer to the network.
IConnectableLayer *	AddFullyConnectedLayer (const FullyConnectedDescriptor &fullyConnectedDescriptor, const char *name=nullptr)
	Adds a fully connected layer to the network.
IConnectableLayer *	AddFusedLayer (const FusedDescriptor &fusedDescriptor, const char *name=nullptr)
	Adds a Fused layer to the network.
IConnectableLayer *	AddPermuteLayer (const PermuteDescriptor &permuteDescriptor, const char *name=nullptr)
	Adds a permute layer to the network.
IConnectableLayer *	AddBatchToSpaceNdLayer (const BatchToSpaceNdDescriptor &batchToSpaceNdDescriptor, const char *name=nullptr)
	Adds a batch to space ND layer to the network.
IConnectableLayer *	AddPooling2dLayer (const Pooling2dDescriptor &pooling2dDescriptor, const char *name=nullptr)
	Adds a 2D pooling layer to the network.
IConnectableLayer *	AddPooling3dLayer (const Pooling3dDescriptor &pooling3dDescriptor, const char *name=nullptr)
	Adds a 3D pooling layer to the network.
IConnectableLayer *	AddPrecompiledLayer (const PreCompiledDescriptor &preCompiledDescriptor, CompiledBlobPtr compiledBlobPtr, const Optional< BackendId > &backend, const char *name=nullptr)
	Adds a Precompiled layer to the network.
IConnectableLayer *	AddActivationLayer (const ActivationDescriptor &activationDescriptor, const char *name=nullptr)
	Adds an activation layer to the network.
IConnectableLayer *	AddNormalizationLayer (const NormalizationDescriptor &normalizationDescriptor, const char *name=nullptr)
	Adds a normalization layer to the network.
IConnectableLayer *	AddSliceLayer (const SliceDescriptor &sliceDescriptor, const char *name=nullptr)
	Adds a slice layer to the network.
IConnectableLayer *	AddSoftmaxLayer (const SoftmaxDescriptor &softmaxDescriptor, const char *name=nullptr)
	Adds a softmax layer to the network.
IConnectableLayer *	AddSplitterLayer (const ViewsDescriptor &splitterDescriptor, const char *name=nullptr)
	Adds a splitter layer to the network.
IConnectableLayer *	AddMergeLayer (const char *name=nullptr)
	Adds a merge layer to the network.
IConnectableLayer *	AddAdditionLayer (const char *name=nullptr)
	Adds an addition layer to the network.
IConnectableLayer *	AddMultiplicationLayer (const char *name=nullptr)
	Adds a multiplication layer to the network.
IConnectableLayer *	AddBatchNormalizationLayer (const BatchNormalizationDescriptor &desc, const ConstTensor &mean, const ConstTensor &variance, const ConstTensor &beta, const ConstTensor &gamma, const char *name=nullptr)
	Adds a batch normalization layer to the network.
IConnectableLayer *	AddRankLayer (const char *name=nullptr)
	Adds a rank layer to the network.
IConnectableLayer *	AddResizeLayer (const ResizeDescriptor &resizeDescriptor, const char *name=nullptr)
	Adds a resize layer to the network.
IConnectableLayer *	AddReduceLayer (const ReduceDescriptor &reduceDescriptor, const char *name=nullptr)
	Adds a reduce layer to the network.
IConnectableLayer *	AddInstanceNormalizationLayer (const InstanceNormalizationDescriptor &desc, const char *name=nullptr)
	Adds an instance normalization layer to the network.
IConnectableLayer *	AddL2NormalizationLayer (const L2NormalizationDescriptor &desc, const char *name=nullptr)
	Adds an L2 normalization layer to the network.
IConnectableLayer *	AddLogSoftmaxLayer (const LogSoftmaxDescriptor &logSoftmaxDescriptor, const char *name=nullptr)
	Adds a log softmax layer to the network.
IConnectableLayer *	AddConstantLayer (const ConstTensor &input, const char *name=nullptr)
	Adds a layer with no inputs and a single output, which always corresponds to the passed in constant tensor.
IConnectableLayer *	AddReshapeLayer (const ReshapeDescriptor &reshapeDescriptor, const char *name=nullptr)
	Adds a reshape layer to the network.
IConnectableLayer *	AddShapeLayer (const char *name=nullptr)
	Adds a shape layer to the network.
IConnectableLayer *	AddSpaceToBatchNdLayer (const SpaceToBatchNdDescriptor &spaceToBatchNdDescriptor, const char *name=nullptr)
	Adds a space to batch layer to the network.
IConnectableLayer *	AddSpaceToDepthLayer (const SpaceToDepthDescriptor &spaceToDepthDescriptor, const char *name=nullptr)
	Adds a space to depth layer to the network.
IConnectableLayer *	AddFloorLayer (const char *name=nullptr)
	Adds a floor layer to the network.
IConnectableLayer *	AddOutputLayer (LayerBindingId id, const char *name=nullptr)
	Adds an output layer to the network.
IConnectableLayer *	AddLstmLayer (const LstmDescriptor &descriptor, const LstmInputParams &params, const char *name=nullptr)
	Add a Lstm layer to the network.
IConnectableLayer *	AddDivisionLayer (const char *name=nullptr)
	Adds a division layer to the network.
IConnectableLayer *	AddSubtractionLayer (const char *name=nullptr)
	Adds a subtraction layer to the network.
IConnectableLayer *	AddMaximumLayer (const char *name=nullptr)
	Add a Maximum layer to the network.
IConnectableLayer *	AddMeanLayer (const MeanDescriptor &meanDescriptor, const char *name=nullptr)
	Add a Mean layer to the network.
IConnectableLayer *	AddPadLayer (const PadDescriptor &padDescriptor, const char *name=nullptr)
	Adds a fully pad layer to the network.
IConnectableLayer *	AddQuantizeLayer (const char *name=nullptr)
	Add a quantize layer to the network.
IConnectableLayer *	AddStridedSliceLayer (const StridedSliceDescriptor &stridedSliceDescriptor, const char *name=nullptr)
	Adds a strided slice layer to the network.
IConnectableLayer *	AddMinimumLayer (const char *name=nullptr)
	Add a Minimum layer to the network.
IConnectableLayer *	AddGatherLayer (const GatherDescriptor &descriptor, const char *name=nullptr)
	Add Gather layer to the network.
IConnectableLayer *	AddGatherNdLayer (const char *name=nullptr)
	Add GatherNd layer to the network.
IConnectableLayer *	AddSwitchLayer (const char *name=nullptr)
	Adds a switch layer to the network.
IConnectableLayer *	AddPreluLayer (const char *name=nullptr)
	Adds a PReLU layer to the network.
IConnectableLayer *	AddTransposeConvolution2dLayer (const TransposeConvolution2dDescriptor &descriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr)
	Adds a 2D transpose convolution layer to the network.
IConnectableLayer *	AddTransposeLayer (const TransposeDescriptor &transposeDescriptor, const char *name=nullptr)
	Adds a transpose layer to the network.
IConnectableLayer *	AddStackLayer (const StackDescriptor &descriptor, const char *name=nullptr)
	Adds a stack layer to the network.
IConnectableLayer *	AddStandInLayer (const StandInDescriptor &descriptor, const char *name=nullptr)
	Add a stand-in layer for a type unknown to the Arm NN framework.
IConnectableLayer *	AddQuantizedLstmLayer (const QuantizedLstmInputParams &params, const char *name=nullptr)
	Add a QuantizedLstm layer to the network.
IConnectableLayer *	AddQLstmLayer (const QLstmDescriptor &descriptor, const LstmInputParams &params, const char *name=nullptr)
	Add a QLstm layer to the network.
IConnectableLayer *	AddLogicalBinaryLayer (const LogicalBinaryDescriptor &descriptor, const char *name=nullptr)
	Adds a Logical Binary layer to the network.
IConnectableLayer *	AddUnidirectionalSequenceLstmLayer (const UnidirectionalSequenceLstmDescriptor &descriptor, const LstmInputParams &params, const char *name=nullptr)
	Add a UnidirectionalSequenceLstm layer to the network.
IConnectableLayer *	AddChannelShuffleLayer (const ChannelShuffleDescriptor &descriptor, const char *name=nullptr)
	Add a ChannelShuffle layer to the network.
IConnectableLayer *	AddBatchMatMulLayer (const BatchMatMulDescriptor &descriptor, const char *name=nullptr)
	Add a BatchMatMul layer to the network.
IConnectableLayer *	AddReverseV2Layer (const char *name=nullptr)
	Add a ReverseV2 layer to the network.
IConnectableLayer *	AddTileLayer (const TileDescriptor &descriptor, const char *name=nullptr)
	Add a Tile layer to the network.
IConnectableLayer *	AddBroadcastToLayer (const BroadcastToDescriptor &descriptor, const char *name=nullptr)
	Add a BroadcastTo layer to the network.
IConnectableLayer *	AddScatterNdLayer (const ScatterNdDescriptor &descriptor, const char *name=nullptr)
	Add a ScatterNd layer to the network.
void	ExecuteStrategy (IStrategy &strategy) const

Static Public Member Functions
static INetwork *	CreateRaw (const NetworkOptions &networkOptions={})
static INetworkPtr	Create (const NetworkOptions &networkOptions={})
static void	Destroy (INetwork *network)

Protected Member Functions
	~INetwork ()
	INetwork (NetworkOptions networkOptions={})

Protected Attributes
std::unique_ptr< NetworkImpl >	pNetworkImpl

Friends
class	TestConnectionPreservation
void	VisitLayersTopologically (const INetwork *inputNetwork, IStrategy &strategy)
TensorInfo	GetInputTensorInfo (const INetwork *network)
IOptimizedNetworkPtr	Optimize (const INetwork &network, const std::vector< BackendId > &backendPreferences, const IDeviceSpec &deviceSpec, const OptimizerOptions &options, Optional< std::vector< std::string > & > messages=EmptyOptional())
	Accept legacy OptimizerOptions.
IOptimizedNetworkPtr	Optimize (const INetwork &network, const std::vector< BackendId > &backendPreferences, const IDeviceSpec &deviceSpec, const OptimizerOptionsOpaque &options=OptimizerOptionsOpaque(), Optional< std::vector< std::string > & > messages=EmptyOptional())
	Create an optimized version of the network.

Detailed Description

Main network class which provides the interface for building up a neural network.

This object is subsequently required by the IRuntime::Load() method.

Definition at line 347 of file INetwork.hpp.

Constructor & Destructor Documentation

~INetwork()

~INetwork ( )

protecteddefault

INetwork()

INetwork ( NetworkOptions networkOptions = {} )

protected

Definition at line 45 of file Network.cpp.

: pNetworkImpl(new NetworkImpl(networkOptions)) {}

References pNetworkImpl.

Referenced by CreateRaw(), Destroy(), GetInputTensorInfo, Optimize, Optimize, and VisitLayersTopologically.

Member Function Documentation

AddActivationLayer()

IConnectableLayer * AddActivationLayer	(	const ActivationDescriptor &	activationDescriptor,
		const char *	name = nullptr )

Adds an activation layer to the network.

Parameters

activationDescriptor	- ActivationDescriptor to configure the activation.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 376 of file Network.cpp.

{
    return pNetworkImpl->AddActivationLayer(activationDescriptor, name);
}

References pNetworkImpl.

AddAdditionLayer()

IConnectableLayer * AddAdditionLayer

(

const char *

name = nullptr

)

Adds an addition layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 409 of file Network.cpp.

{
    ARMNN_NO_DEPRECATE_WARN_BEGIN
    return pNetworkImpl->AddAdditionLayer(name);
    ARMNN_NO_DEPRECATE_WARN_END
}

References ARMNN_NO_DEPRECATE_WARN_BEGIN, ARMNN_NO_DEPRECATE_WARN_END, and pNetworkImpl.

Referenced by armnn::FuseAdditionLayer().

AddArgMinMaxLayer()

IConnectableLayer * AddArgMinMaxLayer	(	const ArgMinMaxDescriptor &	desc,
		const char *	name = nullptr )

Adds an ArgMinMax layer to the network.

Parameters

desc	- Parameters for the L2 normalization operation.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 247 of file Network.cpp.

{
    return pNetworkImpl->AddArgMinMaxLayer(desc, name);
}

References pNetworkImpl.

AddBatchMatMulLayer()

IConnectableLayer * AddBatchMatMulLayer	(	const BatchMatMulDescriptor &	descriptor,
		const char *	name = nullptr )

Add a BatchMatMul layer to the network.

Parameters

descriptor	- Parameters for the BatchMatMul operation
name	- Optional name for the layer

Returns

- Interface for configuring the layer

Definition at line 643 of file Network.cpp.

{
    return pNetworkImpl->AddBatchMatMulLayer(descriptor, name);
}

References pNetworkImpl.

AddBatchNormalizationLayer()

IConnectableLayer * AddBatchNormalizationLayer	(	const BatchNormalizationDescriptor &	desc,
		const ConstTensor &	mean,
		const ConstTensor &	variance,
		const ConstTensor &	beta,
		const ConstTensor &	gamma,
		const char *	name = nullptr )

Adds a batch normalization layer to the network.

Parameters

mean	- Pre-calculated mean for each channel.
variance	- Pre-calculated variance for each channel.
beta	- Per-channel additive factor.
gamma	- Per-channel multiplicative factor.

Returns

- Interface for configuring the layer.

Parameters

name	- Optional name for the layer.

Definition at line 423 of file Network.cpp.

{
    return pNetworkImpl->AddBatchNormalizationLayer(desc, mean, variance, beta, gamma, name);
}

References pNetworkImpl.

Referenced by armnn::FuseBatchNormalizationLayer().

AddBatchToSpaceNdLayer()

IConnectableLayer * AddBatchToSpaceNdLayer	(	const BatchToSpaceNdDescriptor &	batchToSpaceNdDescriptor,
		const char *	name = nullptr )

Adds a batch to space ND layer to the network.

Parameters

batchToSpaceNdDescriptor	- Description of the layer.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 350 of file Network.cpp.

{
    return pNetworkImpl->AddBatchToSpaceNdLayer(batchToSpaceNdDescriptor, name);
}

References pNetworkImpl.

AddBroadcastToLayer()

IConnectableLayer * AddBroadcastToLayer	(	const BroadcastToDescriptor &	descriptor,
		const char *	name = nullptr )

Add a BroadcastTo layer to the network.

Parameters

descriptor	- Parameters for the BroadcastTo operation
name	- Optional name for the layer

Returns

- Interface for configuring the layer

Definition at line 660 of file Network.cpp.

{
    return pNetworkImpl->AddBroadcastToLayer(descriptor, name);
}

References pNetworkImpl.

AddCastLayer()

IConnectableLayer * AddCastLayer

(

const char *

name = nullptr

)

Adds a cast layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 253 of file Network.cpp.

{
    return pNetworkImpl->AddCastLayer(name);
}

References pNetworkImpl.

AddChannelShuffleLayer()

IConnectableLayer * AddChannelShuffleLayer	(	const ChannelShuffleDescriptor &	descriptor,
		const char *	name = nullptr )

Add a ChannelShuffle layer to the network.

Parameters

descriptor	- Parameters for the ChannelShuffle operation
name	- Optional name for the layer

Returns

- Interface for configuring the layer

Definition at line 637 of file Network.cpp.

{
    return pNetworkImpl->AddChannelShuffleLayer(descriptor, name);
}

References pNetworkImpl.

AddComparisonLayer()

IConnectableLayer * AddComparisonLayer	(	const ComparisonDescriptor &	comparisonDescriptor,
		const char *	name = nullptr )

Add a Comparison layer to the network.

Parameters

name	- Optional name for the layer.
desc	- Descriptor for the comparison operation.

Returns

- Interface for configuring the layer.

Definition at line 258 of file Network.cpp.

{
    return pNetworkImpl->AddComparisonLayer(comparisonDescriptor, name);
}

References pNetworkImpl.

AddConcatLayer()

IConnectableLayer * AddConcatLayer	(	const ConcatDescriptor &	concatDescriptor,
		const char *	name = nullptr )

Adds a concatenation layer to the network.

Parameters

concatDescriptor	- ConcatDescriptor (synonym for OriginsDescriptor) to configure the concatenation process. Number of Views must be equal to the number of inputs, and their order must match - e.g. first view corresponds to the first input, second view to the second input, etc....
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 265 of file Network.cpp.

{
    return pNetworkImpl->AddConcatLayer(concatDescriptor, name);
}

References pNetworkImpl.

AddConstantLayer()

IConnectableLayer * AddConstantLayer	(	const ConstTensor &	input,
		const char *	name = nullptr )

Adds a layer with no inputs and a single output, which always corresponds to the passed in constant tensor.

Parameters

input	- Tensor to be provided as the only output of the layer. The layer will maintain its own copy of the tensor data, meaning the memory referenced by input can be freed or reused after this function is called.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 468 of file Network.cpp.

{
    return pNetworkImpl->AddConstantLayer(input, name);
}

References pNetworkImpl.

AddConvolution2dLayer()

IConnectableLayer * AddConvolution2dLayer	(	const Convolution2dDescriptor &	convolution2dDescriptor,
		const char *	name = nullptr )

Adds a 2D convolution layer to the network.

Parameters

convolution2dDescriptor	- Description of the 2D convolution layer.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 272 of file Network.cpp.

{
    return pNetworkImpl->AddConvolution2dLayer(convolution2dDescriptor, name);
}

References pNetworkImpl.

Referenced by armnn::FoldPadLayer2d(), and armnn::FuseConvolution2dLayer().

AddConvolution3dLayer()

IConnectableLayer * AddConvolution3dLayer	(	const Convolution3dDescriptor &	convolution3dDescriptor,
		const char *	name = nullptr )

Adds a 3D convolution layer to the network.

Parameters

convolution3dDescriptor	- Description of the 3D convolution layer.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 278 of file Network.cpp.

{
    return pNetworkImpl->AddConvolution3dLayer(convolution3dDescriptor, name);
}

References pNetworkImpl.

AddDepthToSpaceLayer()

IConnectableLayer * AddDepthToSpaceLayer	(	const DepthToSpaceDescriptor &	depthToSpaceDescriptor,
		const char *	name = nullptr )

Adds a depth to space layer to the network.

Parameters

depthToSpaceDescriptor	- Parameters for the depth to space operation.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 285 of file Network.cpp.

{
    return pNetworkImpl->AddDepthToSpaceLayer(depthToSpaceDescriptor, name);
}

References pNetworkImpl.

AddDepthwiseConvolution2dLayer()

IConnectableLayer * AddDepthwiseConvolution2dLayer	(	const DepthwiseConvolution2dDescriptor &	convolution2dDescriptor,
		const char *	name = nullptr )

Adds a 2D depthwise convolution layer to the network.

Parameters

convolution2dDescriptor	- Description of the 2D depthwise convolution layer.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 292 of file Network.cpp.

{
    return pNetworkImpl->AddDepthwiseConvolution2dLayer(convolution2dDescriptor, name);
}

References pNetworkImpl.

Referenced by armnn::FoldPadLayer2d(), and armnn::FuseDepthwiseConvolution2dLayer().

AddDequantizeLayer()

IConnectableLayer * AddDequantizeLayer

(

const char *

name = nullptr

)

Adds a Dequantize layer to the network.

Returns

- Interface for configuring the layer.

Definition at line 300 of file Network.cpp.

{
    return pNetworkImpl->AddDequantizeLayer(name);
}

References pNetworkImpl.

AddDetectionPostProcessLayer()

IConnectableLayer * AddDetectionPostProcessLayer	(	const DetectionPostProcessDescriptor &	descriptor,
		const ConstTensor &	anchors,
		const char *	name = nullptr )

Adds a Detection PostProcess layer to the network.

Parameters

descriptor	- Description of the Detection PostProcess layer.
anchors	- Tensor for anchors.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 306 of file Network.cpp.

{
    return pNetworkImpl->AddDetectionPostProcessLayer(descriptor, anchors, name);
}

References pNetworkImpl.

AddDivisionLayer()

IConnectableLayer * AddDivisionLayer

(

const char *

name = nullptr

)

Adds a division layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 508 of file Network.cpp.

{
    ARMNN_NO_DEPRECATE_WARN_BEGIN
    return pNetworkImpl->AddDivisionLayer(name);
    ARMNN_NO_DEPRECATE_WARN_END
}

References ARMNN_NO_DEPRECATE_WARN_BEGIN, ARMNN_NO_DEPRECATE_WARN_END, and pNetworkImpl.

Referenced by armnn::FuseDivisionLayer().

AddElementwiseBinaryLayer()

IConnectableLayer * AddElementwiseBinaryLayer	(	const ElementwiseBinaryDescriptor &	elementwiseBinaryDescriptor,
		const char *	name = nullptr )

Add an ElementwiseBinary layer to the network.

Parameters

name	- Optional name for the layer.
desc	- Descriptor for the elementwiseBinary operations.

Returns

- Interface for configuring the layer.

Definition at line 314 of file Network.cpp.

{
    return pNetworkImpl->AddElementwiseBinaryLayer(elementwiseBinaryDescriptor, name);
}

References pNetworkImpl.

Referenced by armnn::FuseElementwiseBinaryLayer().

AddElementwiseUnaryLayer()

IConnectableLayer * AddElementwiseUnaryLayer	(	const ElementwiseUnaryDescriptor &	elementwiseUnaryDescriptor,
		const char *	name = nullptr )

Add an ElementwiseUnary layer to the network.

Parameters

name	- Optional name for the layer.
desc	- Descriptor for the elementwiseUnary operations.

Returns

- Interface for configuring the layer.

Definition at line 320 of file Network.cpp.

{
    return pNetworkImpl->AddElementwiseUnaryLayer(elementwiseUnaryDescriptor, name);
}

References pNetworkImpl.

AddFillLayer()

IConnectableLayer * AddFillLayer	(	const FillDescriptor &	fillDescriptor,
		const char *	name = nullptr )

Add an Fill layer to the network.

Parameters

name	- Optional name for the layer.
fillDescriptor	- Descriptor for the fill operation.

Returns

- Interface for configuring the layer.

Definition at line 326 of file Network.cpp.

{
    return pNetworkImpl->AddFillLayer(fillDescriptor, name);
}

References pNetworkImpl.

AddFloorLayer()

IConnectableLayer * AddFloorLayer

(

const char *

name = nullptr

)

Adds a floor layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 492 of file Network.cpp.

{
    return pNetworkImpl->AddFloorLayer(name);
}

References pNetworkImpl.

AddFullyConnectedLayer()

IConnectableLayer * AddFullyConnectedLayer	(	const FullyConnectedDescriptor &	fullyConnectedDescriptor,
		const char *	name = nullptr )

Adds a fully connected layer to the network.

Parameters

fullyConnectedDescriptor

- Description of the fully connected layer.

Returns

- Interface for configuring the layer.

Note

Weights and biases are passed in as inputs. If they are constant tensors you can simply store them in a ConstantLayer as seen below. A full example can be found in samples/SimpleSample.cpp.

// Make sure the IsConstant flag is set on the weightsInfo before passing it to the ConstTensor.
ConstTensor weights(weightsInfo, weightsData);
 
// Constant layer that now holds weights data for FullyConnected
IConnectableLayer* const constantWeightsLayer = myNetwork->AddConstantLayer(weights, "weights");
 
FullyConnectedDescriptor fullyConnectedDesc;
IConnectableLayer* const fullyConnectedLayer = myNetwork->AddFullyConnectedLayer(fullyConnectedDesc,
                                                                                 "fully connected");
IConnectableLayer* InputLayer = myNetwork->AddInputLayer(0);
InputLayer->GetOutputSlot(0).Connect(fullyConnectedLayer->GetInputSlot(0));
constantWeightsLayer->GetOutputSlot(0).Connect(fullyConnectedLayer->GetInputSlot(1));

Definition at line 332 of file Network.cpp.

{
    return pNetworkImpl->AddFullyConnectedLayer(fullyConnectedDescriptor, name);
}

References pNetworkImpl.

Referenced by armnn::FuseFullyConnectedLayer().

AddFusedLayer()

IConnectableLayer * AddFusedLayer	(	const FusedDescriptor &	fusedDescriptor,
		const char *	name = nullptr )

Adds a Fused layer to the network.

Method use is for backend users.

Parameters

fusedDescriptor	- FusedDescriptor contains parameters for the Fused layer.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 338 of file Network.cpp.

{
    return pNetworkImpl->AddFusedLayer(fusedDescriptor, name);
}

References pNetworkImpl.

Referenced by NeonBackend::OptimizeSubgraphView().

AddGatherLayer()

IConnectableLayer * AddGatherLayer	(	const GatherDescriptor &	descriptor,
		const char *	name = nullptr )

Add Gather layer to the network.

Parameters

descriptor	- Description of the gather layer.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 558 of file Network.cpp.

{
    return pNetworkImpl->AddGatherLayer(descriptor, name);
}

References pNetworkImpl.

AddGatherNdLayer()

IConnectableLayer * AddGatherNdLayer

(

const char *

name = nullptr

)

Add GatherNd layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 564 of file Network.cpp.

{
    return pNetworkImpl->AddGatherNdLayer(name);
}

References pNetworkImpl.

AddInputLayer()

IConnectableLayer * AddInputLayer	(	LayerBindingId	id,
		const char *	name = nullptr )

Adds an input layer to the network.

Parameters

id	- User generated id to uniquely identify a particular input. The same id needs to be specified. when passing the inputs to the IRuntime::EnqueueWorkload() function.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 242 of file Network.cpp.

{
    return pNetworkImpl->AddInputLayer(id, name);
}

References pNetworkImpl.

AddInstanceNormalizationLayer()

IConnectableLayer * AddInstanceNormalizationLayer	(	const InstanceNormalizationDescriptor &	desc,
		const char *	name = nullptr )

Adds an instance normalization layer to the network.

Parameters

desc	- Parameters for the instance normalization operation.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 450 of file Network.cpp.

{
    return pNetworkImpl->AddInstanceNormalizationLayer(desc, name);
}

References pNetworkImpl.

AddL2NormalizationLayer()

IConnectableLayer * AddL2NormalizationLayer	(	const L2NormalizationDescriptor &	desc,
		const char *	name = nullptr )

Adds an L2 normalization layer to the network.

Normalization is performed along dimension 1, but requires a 4d input.

Parameters

desc	- Parameters for the L2 normalization operation.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 456 of file Network.cpp.

{
    return pNetworkImpl->AddL2NormalizationLayer(desc, name);
}

References pNetworkImpl.

AddLogicalBinaryLayer()

IConnectableLayer * AddLogicalBinaryLayer	(	const LogicalBinaryDescriptor &	descriptor,
		const char *	name = nullptr )

Adds a Logical Binary layer to the network.

Parameters

descriptor	- Description of the Logical Binary layer.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 623 of file Network.cpp.

{
    return pNetworkImpl->AddLogicalBinaryLayer(descriptor, name);
}

References pNetworkImpl.

AddLogSoftmaxLayer()

IConnectableLayer * AddLogSoftmaxLayer	(	const LogSoftmaxDescriptor &	logSoftmaxDescriptor,
		const char *	name = nullptr )

Adds a log softmax layer to the network.

Parameters

logSoftmaxDescriptor	- LogSoftmaxDescriptor to configure the log softmax.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 462 of file Network.cpp.

{
    return pNetworkImpl->AddLogSoftmaxLayer(logSoftmaxDescriptor, name);
}

References pNetworkImpl.

AddLstmLayer()

IConnectableLayer * AddLstmLayer	(	const LstmDescriptor &	descriptor,
		const LstmInputParams &	params,
		const char *	name = nullptr )

Add a Lstm layer to the network.

Parameters

descriptor	- Parameters for the Lstm operation
params	- Weights and biases for the LSTM cell
name	- Optional name for the layer

Returns

- Interface for configuring the layer.

Definition at line 501 of file Network.cpp.

{
    return pNetworkImpl->AddLstmLayer(descriptor, params, name);
}

References pNetworkImpl.

AddMaximumLayer()

IConnectableLayer * AddMaximumLayer

(

const char *

name = nullptr

)

Add a Maximum layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 522 of file Network.cpp.

{
    ARMNN_NO_DEPRECATE_WARN_BEGIN
    return pNetworkImpl->AddMaximumLayer(name);
    ARMNN_NO_DEPRECATE_WARN_END
}

References ARMNN_NO_DEPRECATE_WARN_BEGIN, ARMNN_NO_DEPRECATE_WARN_END, and pNetworkImpl.

AddMeanLayer()

IConnectableLayer * AddMeanLayer	(	const MeanDescriptor &	meanDescriptor,
		const char *	name = nullptr )

Add a Mean layer to the network.

Parameters

meanDescriptor	- Parameters for the mean operation.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 529 of file Network.cpp.

{
    return pNetworkImpl->AddMeanLayer(meanDescriptor, name);
}

References pNetworkImpl.

AddMergeLayer()

IConnectableLayer * AddMergeLayer

(

const char *

name = nullptr

)

Adds a merge layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 404 of file Network.cpp.

{
    return pNetworkImpl->AddMergeLayer(name);
}

References pNetworkImpl.

AddMinimumLayer()

IConnectableLayer * AddMinimumLayer

(

const char *

name = nullptr

)

Add a Minimum layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 551 of file Network.cpp.

{
    ARMNN_NO_DEPRECATE_WARN_BEGIN
    return pNetworkImpl->AddMinimumLayer(name);
    ARMNN_NO_DEPRECATE_WARN_END
}

References ARMNN_NO_DEPRECATE_WARN_BEGIN, ARMNN_NO_DEPRECATE_WARN_END, and pNetworkImpl.

AddMultiplicationLayer()

IConnectableLayer * AddMultiplicationLayer

(

const char *

name = nullptr

)

Adds a multiplication layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 416 of file Network.cpp.

{
    ARMNN_NO_DEPRECATE_WARN_BEGIN
    return pNetworkImpl->AddMultiplicationLayer(name);
    ARMNN_NO_DEPRECATE_WARN_END
}

References ARMNN_NO_DEPRECATE_WARN_BEGIN, ARMNN_NO_DEPRECATE_WARN_END, and pNetworkImpl.

Referenced by armnn::FuseMultiplicationLayer().

AddNormalizationLayer()

IConnectableLayer * AddNormalizationLayer	(	const NormalizationDescriptor &	normalizationDescriptor,
		const char *	name = nullptr )

Adds a normalization layer to the network.

Parameters

normalizationDescriptor	- NormalizationDescriptor to configure the normalization.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 382 of file Network.cpp.

{
    return pNetworkImpl->AddNormalizationLayer(normalizationDescriptor, name);
}

References pNetworkImpl.

AddOutputLayer()

IConnectableLayer * AddOutputLayer	(	LayerBindingId	id,
		const char *	name = nullptr )

Adds an output layer to the network.

Parameters

id	- User generated id to uniquely identify a particular output. The same id needs to be specified when passing the outputs to the IRuntime::EnqueueWorkload() function.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 496 of file Network.cpp.

{
    return pNetworkImpl->AddOutputLayer(id, name);
}

References pNetworkImpl.

AddPadLayer()

IConnectableLayer * AddPadLayer	(	const PadDescriptor &	padDescriptor,
		const char *	name = nullptr )

Adds a fully pad layer to the network.

Parameters

paddings	- n by 2 tensor, where n is the rank of the input tensor, such that paddings[i,0] indicates the amount of padding to add in front of dimonsion i, and paddings[i,1] indicates the amount of padding to add after the end of dimension i
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 534 of file Network.cpp.

{
    return pNetworkImpl->AddPadLayer(padDescriptor, name);
}

References pNetworkImpl.

AddPermuteLayer()

IConnectableLayer * AddPermuteLayer	(	const PermuteDescriptor &	permuteDescriptor,
		const char *	name = nullptr )

Adds a permute layer to the network.

Parameters

permuteDescriptor	- PermuteDescriptor to configure the permute.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 344 of file Network.cpp.

{
    return pNetworkImpl->AddPermuteLayer(permuteDescriptor, name);
}

References pNetworkImpl.

AddPooling2dLayer()

IConnectableLayer * AddPooling2dLayer	(	const Pooling2dDescriptor &	pooling2dDescriptor,
		const char *	name = nullptr )

Adds a 2D pooling layer to the network.

Parameters

pooling2dDescriptor	- Pooling2dDescriptor to configure the pooling.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 356 of file Network.cpp.

{
    return pNetworkImpl->AddPooling2dLayer(pooling2dDescriptor, name);
}

References pNetworkImpl.

Referenced by armnn::FoldPadLayer2d().

AddPooling3dLayer()

IConnectableLayer * AddPooling3dLayer	(	const Pooling3dDescriptor &	pooling3dDescriptor,
		const char *	name = nullptr )

Adds a 3D pooling layer to the network.

Parameters

pooling3dDescriptor	- Pooling3dDescriptor to configure the pooling.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 362 of file Network.cpp.

{
    return pNetworkImpl->AddPooling3dLayer(pooling3dDescriptor, name);
}

References pNetworkImpl.

AddPrecompiledLayer()

IConnectableLayer * AddPrecompiledLayer	(	const PreCompiledDescriptor &	preCompiledDescriptor,
		CompiledBlobPtr	compiledBlobPtr,
		const Optional< BackendId > &	backend,
		const char *	name = nullptr )

Adds a Precompiled layer to the network.

Method use is for backend users.

Parameters

preCompiledDescriptor	- PreCompiledDescriptor contains parameters for the Precompiled layer.
compiledBlobPtr	- CompiledBlobPtr pre-compiled object set for the Precompiled layer.
backend	- optional BackendId set for the Precompiled layer.

Returns

- Interface for configuring the layer.

Definition at line 368 of file Network.cpp.

{
    return pNetworkImpl->AddPrecompiledLayer(preCompiledDescriptor, std::move(compiledBlobPtr), backend, name);
}

References pNetworkImpl.

Referenced by TosaRefBackend::OptimizeSubgraphView().

AddPreluLayer()

IConnectableLayer * AddPreluLayer

(

const char *

name = nullptr

)

Adds a PReLU layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 574 of file Network.cpp.

{
    return pNetworkImpl->AddPreluLayer(name);
}

References pNetworkImpl.

AddQLstmLayer()

IConnectableLayer * AddQLstmLayer	(	const QLstmDescriptor &	descriptor,
		const LstmInputParams &	params,
		const char *	name = nullptr )

Add a QLstm layer to the network.

Parameters

descriptor	- Parameters for the QLstm operation
params	- Weights and biases for the layer
name	- Optional name for the layer

Returns

- Interface for configuring the layer.

Definition at line 616 of file Network.cpp.

{
    return pNetworkImpl->AddQLstmLayer(descriptor, params, name);
}

References pNetworkImpl.

AddQuantizedLstmLayer()

IConnectableLayer * AddQuantizedLstmLayer	(	const QuantizedLstmInputParams &	params,
		const char *	name = nullptr )

Add a QuantizedLstm layer to the network.

Parameters

params	- The weights and biases for the Quantized LSTM cell
name	- Optional name for the layer

Returns

- Interface for configuring the layer.

Definition at line 610 of file Network.cpp.

{
    return pNetworkImpl->AddQuantizedLstmLayer(params, name);
}

References pNetworkImpl.

AddQuantizeLayer()

IConnectableLayer * AddQuantizeLayer

(

const char *

name = nullptr

)

Add a quantize layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 540 of file Network.cpp.

{
    return pNetworkImpl->AddQuantizeLayer(name);
}

References pNetworkImpl.

AddRankLayer()

IConnectableLayer * AddRankLayer

(

const char *

name = nullptr

)

Adds a rank layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 433 of file Network.cpp.

{
    return pNetworkImpl->AddRankLayer(name);
}

References pNetworkImpl.

AddReduceLayer()

IConnectableLayer * AddReduceLayer	(	const ReduceDescriptor &	reduceDescriptor,
		const char *	name = nullptr )

Adds a reduce layer to the network.

Parameters

ReduceDescriptor	- Parameters for the reduce operation.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 444 of file Network.cpp.

{
    return pNetworkImpl->AddReduceLayer(reduceDescriptor, name);
}

References pNetworkImpl.

Referenced by armnn::ChainReduceLayers().

AddReshapeLayer()

IConnectableLayer * AddReshapeLayer	(	const ReshapeDescriptor &	reshapeDescriptor,
		const char *	name = nullptr )

Adds a reshape layer to the network.

Parameters

reshapeDescriptor	- Parameters for the reshape operation.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 474 of file Network.cpp.

{
    return pNetworkImpl->AddReshapeLayer(reshapeDescriptor, name);
}

References pNetworkImpl.

AddResizeLayer()

IConnectableLayer * AddResizeLayer	(	const ResizeDescriptor &	resizeDescriptor,
		const char *	name = nullptr )

Adds a resize layer to the network.

Parameters

resizeDescriptor	- Parameters for the resize operation.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 438 of file Network.cpp.

{
    return pNetworkImpl->AddResizeLayer(resizeDescriptor, name);
}

References pNetworkImpl.

AddReverseV2Layer()

IConnectableLayer * AddReverseV2Layer

(

const char *

name = nullptr

)

Add a ReverseV2 layer to the network.

Parameters

name	- Optional name for the layer

Returns

- Interface for configuring the layer

Definition at line 649 of file Network.cpp.

{
    return pNetworkImpl->AddReverseV2Layer(name);
}

References pNetworkImpl.

AddScatterNdLayer()

IConnectableLayer * AddScatterNdLayer	(	const ScatterNdDescriptor &	descriptor,
		const char *	name = nullptr )

Add a ScatterNd layer to the network.

Parameters

descriptor	- Parameters for the ScatterNd operation
name	- Optional name for the layer

Returns

- Interface for configuring the layer

Definition at line 666 of file Network.cpp.

{
    return pNetworkImpl->AddScatterNdLayer(descriptor, name);
}

References pNetworkImpl.

AddShapeLayer()

IConnectableLayer * AddShapeLayer

(

const char *

name = nullptr

)

Adds a shape layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 593 of file Network.cpp.

{
    return pNetworkImpl->AddShapeLayer(name);
}

References pNetworkImpl.

AddSliceLayer()

IConnectableLayer * AddSliceLayer	(	const SliceDescriptor &	sliceDescriptor,
		const char *	name = nullptr )

Adds a slice layer to the network.

Parameters

sliceDescriptor	- SliceDescriptor to configure the slice operation.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 388 of file Network.cpp.

{
    return pNetworkImpl->AddSliceLayer(sliceDescriptor, name);
}

References pNetworkImpl.

AddSoftmaxLayer()

IConnectableLayer * AddSoftmaxLayer	(	const SoftmaxDescriptor &	softmaxDescriptor,
		const char *	name = nullptr )

Adds a softmax layer to the network.

If the data type is QAsymm8, then the output quantization parameters must have a scale of 1/256 and an offset of 0

Parameters

softmaxDescriptor	- SoftmaxDescriptor to configure the softmax.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 392 of file Network.cpp.

{
    return pNetworkImpl->AddSoftmaxLayer(softmaxDescriptor, name);
}

References pNetworkImpl.

AddSpaceToBatchNdLayer()

IConnectableLayer * AddSpaceToBatchNdLayer	(	const SpaceToBatchNdDescriptor &	spaceToBatchNdDescriptor,
		const char *	name = nullptr )

Adds a space to batch layer to the network.

Parameters

spaceToBatchNdDescriptor	- Parameters for the space to batch operation.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 480 of file Network.cpp.

{
    return pNetworkImpl->AddSpaceToBatchNdLayer(spaceToBatchNdDescriptor, name);
}

References pNetworkImpl.

AddSpaceToDepthLayer()

IConnectableLayer * AddSpaceToDepthLayer	(	const SpaceToDepthDescriptor &	spaceToDepthDescriptor,
		const char *	name = nullptr )

Adds a space to depth layer to the network.

Parameters

spaceToDepthDescriptor	- Parameters for the space to depth operation.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 486 of file Network.cpp.

{
    return pNetworkImpl->AddSpaceToDepthLayer(spaceToDepthDescriptor, name);
}

References pNetworkImpl.

AddSplitterLayer()

IConnectableLayer * AddSplitterLayer	(	const ViewsDescriptor &	splitterDescriptor,
		const char *	name = nullptr )

Adds a splitter layer to the network.

Parameters

splitterDescriptor	- ViewsDescriptor to configure the splitting process. Number of Views must be equal to the number of outputs, and their order must match - e.g. first view corresponds to the first output, second view to the second output, etc....
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 398 of file Network.cpp.

{
    return pNetworkImpl->AddSplitterLayer(splitterDescriptor, name);
}

References pNetworkImpl.

AddStackLayer()

IConnectableLayer * AddStackLayer	(	const StackDescriptor &	descriptor,
		const char *	name = nullptr )

Adds a stack layer to the network.

Parameters

descriptor	- Description of the stack layer.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 598 of file Network.cpp.

{
    return pNetworkImpl->AddStackLayer(descriptor, name);
}

References pNetworkImpl.

AddStandInLayer()

IConnectableLayer * AddStandInLayer	(	const StandInDescriptor &	descriptor,
		const char *	name = nullptr )

Add a stand-in layer for a type unknown to the Arm NN framework.

Note: Due to the nature of this layer, no validation can be performed by the framework. Furthermore, Any model containing this layer cannot make use of dynamic tensors since the tensor sizes cannot be inferred. @descriptor - Descriptor for the StandIn layer.

Returns

- Interface for configuring the layer.

Definition at line 604 of file Network.cpp.

{
    return pNetworkImpl->AddStandInLayer(descriptor, name);
}

References pNetworkImpl.

AddStridedSliceLayer()

IConnectableLayer * AddStridedSliceLayer	(	const StridedSliceDescriptor &	stridedSliceDescriptor,
		const char *	name = nullptr )

Adds a strided slice layer to the network.

Parameters

StridedSliceDescriptor	- Parameters for the strided slice operation.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 545 of file Network.cpp.

{
    return pNetworkImpl->AddStridedSliceLayer(stridedSliceDescriptor, name);
}

References pNetworkImpl.

AddSubtractionLayer()

IConnectableLayer * AddSubtractionLayer

(

const char *

name = nullptr

)

Adds a subtraction layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 515 of file Network.cpp.

{
    ARMNN_NO_DEPRECATE_WARN_BEGIN
    return pNetworkImpl->AddSubtractionLayer(name);
    ARMNN_NO_DEPRECATE_WARN_END
}

References ARMNN_NO_DEPRECATE_WARN_BEGIN, ARMNN_NO_DEPRECATE_WARN_END, and pNetworkImpl.

Referenced by armnn::FuseSubtractionLayer().

AddSwitchLayer()

IConnectableLayer * AddSwitchLayer

(

const char *

name = nullptr

)

Adds a switch layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 569 of file Network.cpp.

{
    return pNetworkImpl->AddSwitchLayer(name);
}

References pNetworkImpl.

AddTileLayer()

IConnectableLayer * AddTileLayer	(	const TileDescriptor &	descriptor,
		const char *	name = nullptr )

Add a Tile layer to the network.

Parameters

descriptor	- Parameters for the Tile operation
name	- Optional name for the layer

Returns

- Interface for configuring the layer

Definition at line 654 of file Network.cpp.

{
    return pNetworkImpl->AddTileLayer(descriptor, name);
}

References pNetworkImpl.

AddTransposeConvolution2dLayer()

IConnectableLayer * AddTransposeConvolution2dLayer	(	const TransposeConvolution2dDescriptor &	descriptor,
		const ConstTensor &	weights,
		const Optional< ConstTensor > &	biases,
		const char *	name = nullptr )

Adds a 2D transpose convolution layer to the network.

Parameters

descriptor	- Description of the 2D transpose convolution layer.
weights	- Tensor for the weights data.
biases	- Optional tensor for the bias data.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 579 of file Network.cpp.

{
    return pNetworkImpl->AddTransposeConvolution2dLayer(descriptor, weights, biases, name);
}

References pNetworkImpl.

AddTransposeLayer()

IConnectableLayer * AddTransposeLayer	(	const TransposeDescriptor &	transposeDescriptor,
		const char *	name = nullptr )

Adds a transpose layer to the network.

Parameters

transposeDescriptor	- TransposeDescriptor to configure the transpose.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Definition at line 587 of file Network.cpp.

{
    return pNetworkImpl->AddTransposeLayer(transposeDescriptor, name);
}

References pNetworkImpl.

AddUnidirectionalSequenceLstmLayer()

IConnectableLayer * AddUnidirectionalSequenceLstmLayer	(	const UnidirectionalSequenceLstmDescriptor &	descriptor,
		const LstmInputParams &	params,
		const char *	name = nullptr )

Add a UnidirectionalSequenceLstm layer to the network.

Parameters

descriptor	- Parameters for the UnidirectionalSequenceLstm operation
params	- Weights and biases for the UnidirectionalSequenceLstm
name	- Optional name for the layer

Returns

- Interface for configuring the layer.

Definition at line 629 of file Network.cpp.

{
    return pNetworkImpl->AddUnidirectionalSequenceLstmLayer(descriptor, params, name);
}

References pNetworkImpl.

Create()

armnn::INetworkPtr Create ( const NetworkOptions & networkOptions = {} )

static

Examples

CustomMemoryAllocatorSample.cpp, DynamicSample.cpp, and SimpleSample.cpp.

Definition at line 682 of file Network.cpp.

{
    return INetworkPtr(CreateRaw(networkOptions), &INetwork::Destroy);
}

References CreateRaw(), and Destroy().

CreateRaw()

armnn::INetwork * CreateRaw ( const NetworkOptions & networkOptions = {} )

static

Definition at line 677 of file Network.cpp.

{
    return new INetwork(networkOptions);
}

References INetwork().

Referenced by Create().

Destroy()

void Destroy ( INetwork * network )

static

Definition at line 687 of file Network.cpp.

{
    delete network;
}

References INetwork().

Referenced by Create().

ExecuteStrategy()

void ExecuteStrategy

(

IStrategy &

strategy

)

const

Definition at line 672 of file Network.cpp.

{
    return pNetworkImpl->ExecuteStrategy(strategy);
}

References pNetworkImpl.

Referenced by ISerializer::SerializerImpl::Serialize().

PrintGraph()

Status PrintGraph

(

)

Definition at line 237 of file Network.cpp.

{
    return pNetworkImpl->PrintGraph();
}

References pNetworkImpl.

Friends And Related Symbol Documentation

GetInputTensorInfo

TensorInfo GetInputTensorInfo ( const INetwork * network )

friend

References INetwork().

Optimize [1/2]

IOptimizedNetworkPtr Optimize ( const INetwork & network, const std::vector< BackendId > & backendPreferences, const IDeviceSpec & deviceSpec, const OptimizerOptions & options, Optional< std::vector< std::string > & > messages = EmptyOptional() )

friend

Accept legacy OptimizerOptions.

Definition at line 2274 of file Network.cpp.

{
    return Optimize(inNetwork,
                    backendPreferences,
                    deviceSpec,
                    OptimizerOptionsOpaque(options),
                    messages);
}

References INetwork(), and armnn::Optimize().

Optimize [2/2]

IOptimizedNetworkPtr Optimize ( const INetwork & network, const std::vector< BackendId > & backendPreferences, const IDeviceSpec & deviceSpec, const OptimizerOptionsOpaque & options = OptimizerOptionsOpaque(), Optional< std::vector< std::string > & > messages = EmptyOptional() )

friend

Create an optimized version of the network.

Parameters

network	INetwork description of the network to be optimized.
backendPreferences	The choice of the backend ordered by user preferences.
deviceSpec	DeviceSpec object as queried from the runtime. See IRuntime::GetDeviceSpec()
messages	If there are failures or warnings a string describing same will be added to the vector
options	OptimizerOptions object with optimizer configuration options

Returns

An IOptimizedNetworkPtr interface to the optimized network, throws an exception derived from armnn::Exception if process fails.

Definition at line 2287 of file Network.cpp.

{
    return Optimize(inNetwork.pNetworkImpl->GetGraph(),
                    backendPreferences,
                    deviceSpec,
                    options,
                    messages);
}

References INetwork(), armnn::Optimize(), and pNetworkImpl.

TestConnectionPreservation

friend class TestConnectionPreservation

friend

Definition at line 880 of file INetwork.hpp.

References TestConnectionPreservation.

Referenced by TestConnectionPreservation.

VisitLayersTopologically

void VisitLayersTopologically ( const INetwork * inputNetwork, IStrategy & strategy )

friend

References INetwork().

Member Data Documentation

pNetworkImpl

std::unique_ptr<NetworkImpl> pNetworkImpl

protected

Definition at line 895 of file INetwork.hpp.

Referenced by AddActivationLayer(), AddAdditionLayer(), AddArgMinMaxLayer(), AddBatchMatMulLayer(), AddBatchNormalizationLayer(), AddBatchToSpaceNdLayer(), AddBroadcastToLayer(), AddCastLayer(), AddChannelShuffleLayer(), AddComparisonLayer(), AddConcatLayer(), AddConstantLayer(), AddConvolution2dLayer(), AddConvolution3dLayer(), AddDepthToSpaceLayer(), AddDepthwiseConvolution2dLayer(), AddDequantizeLayer(), AddDetectionPostProcessLayer(), AddDivisionLayer(), AddElementwiseBinaryLayer(), AddElementwiseUnaryLayer(), AddFillLayer(), AddFloorLayer(), AddFullyConnectedLayer(), AddFusedLayer(), AddGatherLayer(), AddGatherNdLayer(), AddInputLayer(), AddInstanceNormalizationLayer(), AddL2NormalizationLayer(), AddLogicalBinaryLayer(), AddLogSoftmaxLayer(), AddLstmLayer(), AddMaximumLayer(), AddMeanLayer(), AddMergeLayer(), AddMinimumLayer(), AddMultiplicationLayer(), AddNormalizationLayer(), AddOutputLayer(), AddPadLayer(), AddPermuteLayer(), AddPooling2dLayer(), AddPooling3dLayer(), AddPrecompiledLayer(), AddPreluLayer(), AddQLstmLayer(), AddQuantizedLstmLayer(), AddQuantizeLayer(), AddRankLayer(), AddReduceLayer(), AddReshapeLayer(), AddResizeLayer(), AddReverseV2Layer(), AddScatterNdLayer(), AddShapeLayer(), AddSliceLayer(), AddSoftmaxLayer(), AddSpaceToBatchNdLayer(), AddSpaceToDepthLayer(), AddSplitterLayer(), AddStackLayer(), AddStandInLayer(), AddStridedSliceLayer(), AddSubtractionLayer(), AddSwitchLayer(), AddTileLayer(), AddTransposeConvolution2dLayer(), AddTransposeLayer(), AddUnidirectionalSequenceLstmLayer(), ExecuteStrategy(), INetwork(), Optimize, and PrintGraph().

---

The documentation for this class was generated from the following files:

• include/armnn/INetwork.hpp
• src/armnn/Network.cpp