Network Class Reference

Private implementation of INetwork. More...

#include <Network.hpp>

Inheritance diagram for Network:

Public Member Functions
	Network ()
	~Network ()
const Graph &	GetGraph () const
Status	PrintGraph () override
profiling::ProfilingGuid	GetGuid () const final
IConnectableLayer *	AddInputLayer (LayerBindingId id, const char *name=nullptr) override
IConnectableLayer *	AddArgMinMaxLayer (const ArgMinMaxDescriptor &desc, const char *name=nullptr) override
IConnectableLayer *	AddBatchToSpaceNdLayer (const BatchToSpaceNdDescriptor &batchToSpaceNdDescriptor, const char *name=nullptr) override
IConnectableLayer *	AddComparisonLayer (const ComparisonDescriptor &comparisonDescriptor, const char *name=nullptr) override
IConnectableLayer *	AddConcatLayer (const ConcatDescriptor &concatDescriptor, const char *name=nullptr) override
IConnectableLayer *	AddConvolution2dLayer (const Convolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr) override
IConnectableLayer *	AddConvolution2dLayer (const Convolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const char *name=nullptr) override
IConnectableLayer *	AddConvolution2dLayer (const Convolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const ConstTensor &biases, const char *name=nullptr) override
IConnectableLayer *	AddDepthToSpaceLayer (const DepthToSpaceDescriptor &depthToSpaceDescriptor, const char *name=nullptr) override
IConnectableLayer *	AddDepthwiseConvolution2dLayer (const DepthwiseConvolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr) override
IConnectableLayer *	AddDepthwiseConvolution2dLayer (const DepthwiseConvolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const char *name=nullptr) override
IConnectableLayer *	AddDepthwiseConvolution2dLayer (const DepthwiseConvolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const ConstTensor &biases, const char *name=nullptr) override
IConnectableLayer *	AddDequantizeLayer (const char *name=nullptr) override
IConnectableLayer *	AddDetectionPostProcessLayer (const DetectionPostProcessDescriptor &descriptor, const ConstTensor &anchors, const char *name=nullptr) override
IConnectableLayer *	AddElementwiseUnaryLayer (const ElementwiseUnaryDescriptor &elementwiseUnaryDescriptor, const char *name=nullptr) override
IConnectableLayer *	AddFullyConnectedLayer (const FullyConnectedDescriptor &fullyConnectedDescriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr) override
IConnectableLayer *	AddFullyConnectedLayer (const FullyConnectedDescriptor &fullyConnectedDescriptor, const ConstTensor &weights, const char *name=nullptr) override
IConnectableLayer *	AddFullyConnectedLayer (const FullyConnectedDescriptor &fullyConnectedDescriptor, const ConstTensor &weights, const ConstTensor &biases, const char *name=nullptr) override
IConnectableLayer *	AddGatherLayer (const char *name=nullptr) override
IConnectableLayer *	AddPermuteLayer (const PermuteDescriptor &permuteDescriptor, const char *name=nullptr) override
IConnectableLayer *	AddPooling2dLayer (const Pooling2dDescriptor &pooling2dDescriptor, const char *name=nullptr) override
IConnectableLayer *	AddActivationLayer (const ActivationDescriptor &activationDescriptor, const char *name=nullptr) override
IConnectableLayer *	AddNormalizationLayer (const NormalizationDescriptor &normalizationDescriptor, const char *name=nullptr) override
IConnectableLayer *	AddSliceLayer (const SliceDescriptor &sliceDescriptor, const char *name=nullptr) override
IConnectableLayer *	AddSoftmaxLayer (const SoftmaxDescriptor &softmaxDescriptor, const char *name=nullptr) override
IConnectableLayer *	AddSplitterLayer (const ViewsDescriptor &splitterDescriptor, const char *name=nullptr) override
IConnectableLayer *	AddMergerLayer (const MergerDescriptor &mergerDescriptor, const char *name=nullptr) override
IConnectableLayer *	AddAbsLayer (const char *name=nullptr) override
IConnectableLayer *	AddAdditionLayer (const char *name=nullptr) override
IConnectableLayer *	AddMultiplicationLayer (const char *name=nullptr) override
IConnectableLayer *	AddBatchNormalizationLayer (const BatchNormalizationDescriptor &desc, const ConstTensor &mean, const ConstTensor &variance, const ConstTensor &beta, const ConstTensor &gamma, const char *name=nullptr) override
IConnectableLayer *	AddResizeBilinearLayer (const ResizeBilinearDescriptor &resizeDesc, const char *name=nullptr) override
IConnectableLayer *	AddResizeLayer (const ResizeDescriptor &resizeDescriptor, const char *name=nullptr) override
IConnectableLayer *	AddInstanceNormalizationLayer (const InstanceNormalizationDescriptor &desc, const char *name=nullptr) override
IConnectableLayer *	AddL2NormalizationLayer (const L2NormalizationDescriptor &desc, const char *name=nullptr) override
IConnectableLayer *	AddLogSoftmaxLayer (const LogSoftmaxDescriptor &logSoftmaxDescriptor, const char *name=nullptr) override
IConnectableLayer *	AddConstantLayer (const ConstTensor &input, const char *name=nullptr) override
IConnectableLayer *	AddReshapeLayer (const ReshapeDescriptor &reshapeDescriptor, const char *name=nullptr) override
IConnectableLayer *	AddSpaceToBatchNdLayer (const SpaceToBatchNdDescriptor &spaceToBatchNdDescriptor, const char *name=nullptr) override
IConnectableLayer *	AddSpaceToDepthLayer (const SpaceToDepthDescriptor &spaceToDepthDescriptor, const char *name=nullptr) override
IConnectableLayer *	AddFloorLayer (const char *name=nullptr) override
IConnectableLayer *	AddOutputLayer (LayerBindingId id, const char *name=nullptr) override
IConnectableLayer *	AddLstmLayer (const LstmDescriptor &descriptor, const LstmInputParams &params, const char *name=nullptr) override
IConnectableLayer *	AddDivisionLayer (const char *name=nullptr) override
IConnectableLayer *	AddSubtractionLayer (const char *name=nullptr) override
IConnectableLayer *	AddMaximumLayer (const char *name=nullptr) override
IConnectableLayer *	AddMeanLayer (const MeanDescriptor &meanDescriptor, const char *name=nullptr) override
IConnectableLayer *	AddPadLayer (const PadDescriptor &padDescriptor, const char *name=nullptr) override
IConnectableLayer *	AddQuantizeLayer (const char *name=nullptr) override
IConnectableLayer *	AddStridedSliceLayer (const StridedSliceDescriptor &stridedSliceDescriptor, const char *name=nullptr) override
IConnectableLayer *	AddMinimumLayer (const char *name=nullptr) override
IConnectableLayer *	AddGreaterLayer (const char *name=nullptr) override
IConnectableLayer *	AddEqualLayer (const char *name=nullptr) override
IConnectableLayer *	AddRsqrtLayer (const char *name=nullptr) override
IConnectableLayer *	AddMergeLayer (const char *name=nullptr) override
IConnectableLayer *	AddSwitchLayer (const char *name=nullptr) override
IConnectableLayer *	AddPreluLayer (const char *name=nullptr) override
IConnectableLayer *	AddTransposeConvolution2dLayer (const TransposeConvolution2dDescriptor &descriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr) override
IConnectableLayer *	AddStackLayer (const StackDescriptor &stackDescriptor, const char *name=nullptr) override
IConnectableLayer *	AddStandInLayer (const StandInDescriptor &descriptor, const char *name=nullptr) override
IConnectableLayer *	AddQuantizedLstmLayer (const QuantizedLstmInputParams &params, const char *name=nullptr) override
void	Accept (ILayerVisitor &visitor) const override

Additional Inherited Members
Static Public Member Functions inherited from INetwork
static INetwork *	CreateRaw ()
static INetworkPtr	Create ()
static void	Destroy (INetwork *network)
Protected Member Functions inherited from INetwork
	~INetwork ()

Detailed Description

Private implementation of INetwork.

Definition at line 27 of file Network.hpp.

Constructor & Destructor Documentation

Network()

Network

(

)

Definition at line 937 of file Network.cpp.

: m_Graph(std::make_unique<Graph>()),
  m_Guid(profiling::ProfilingService::Instance().NextGuid())
{
}

~Network()

~Network

(

)

Definition at line 943 of file Network.cpp.

{
}

Member Function Documentation

Accept()

void Accept ( ILayerVisitor &  visitor ) const

overridevirtual

Implements INetwork.

Definition at line 1580 of file Network.cpp.

References Network::GetGraph().

Referenced by Network::GetGuid().

{
    for (auto layer : GetGraph())
    {
        layer->Accept(visitor);
    };
}

AddAbsLayer()

IConnectableLayer * AddAbsLayer ( const char *  name = nullptr )

overridevirtual

Add absolute layer to the network.

Parameters

name	- Optional name for the layer. @ return - Interface for configuring the layer.

Implements INetwork.

Definition at line 1207 of file Network.cpp.

References armnn::Abs, and Network::AddElementwiseUnaryLayer().

Referenced by Network::GetGuid().

{
    return AddElementwiseUnaryLayer(ElementwiseUnaryDescriptor(UnaryOperation::Abs), name);
}

AddActivationLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1155 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), and Network::GetGuid().

{
    return m_Graph->AddLayer<ActivationLayer>(activationDescriptor, name);
}

AddAdditionLayer()

IConnectableLayer * AddAdditionLayer ( const char *  name = nullptr )

overridevirtual

Adds an addition layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Implements INetwork.

Definition at line 1212 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), and Network::GetGuid().

{
    return m_Graph->AddLayer<AdditionLayer>(name);
}

AddArgMinMaxLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1161 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<ArgMinMaxLayer>(argMinMaxDescriptor, name);
}

AddBatchNormalizationLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1227 of file Network.cpp.

References BatchNormalizationLayer::m_Mean.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), and Network::GetGuid().

{
    const auto layer = m_Graph->AddLayer<BatchNormalizationLayer>(desc, name);

    layer->m_Mean = std::make_unique<ScopedCpuTensorHandle>(mean);
    layer->m_Variance = std::make_unique<ScopedCpuTensorHandle>(variance);
    layer->m_Beta = std::make_unique<ScopedCpuTensorHandle>(beta);
    layer->m_Gamma = std::make_unique<ScopedCpuTensorHandle>(gamma);

    return layer;
}

AddBatchToSpaceNdLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 958 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<BatchToSpaceNdLayer>(batchToSpaceNdDescriptor, name);
}

AddComparisonLayer()

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

overridevirtual

Add a Comparison layer to the network.

Parameters

name	- Optional name for the layer.
desc	- Descriptor for the comparison operation. @ return - Interface for configuring the layer.

Implements INetwork.

Definition at line 964 of file Network.cpp.

Referenced by Network::AddEqualLayer(), Network::AddGreaterLayer(), and Network::GetGuid().

{
    return m_Graph->AddLayer<ComparisonLayer>(comparisonDescriptor, name);
}

AddConcatLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1023 of file Network.cpp.

References OptionalBase::has_value(), Convolution2dDescriptor::m_BiasEnabled, Convolution2dLayer::m_Weight, and OptionalReferenceSwitch< std::is_reference< T >::value, T >::value().

Referenced by Network::AddMergerLayer(), BOOST_AUTO_TEST_CASE(), and Network::GetGuid().

{
    return m_Graph->AddLayer<ConcatLayer>(concatDescriptor, name);
}

AddConstantLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1280 of file Network.cpp.

References ConstantLayer::m_LayerOutput.

Referenced by armnn::BOOST_AUTO_TEST_CASE(), and Network::GetGuid().

{
    auto layer = m_Graph->AddLayer<ConstantLayer>(name);

    layer->m_LayerOutput = std::make_unique<ScopedCpuTensorHandle>(input);

    return layer;
}

AddConvolution2dLayer() [1/3]

IConnectableLayer * AddConvolution2dLayer ( const Convolution2dDescriptor &  convolution2dDescriptor, const ConstTensor &  weights, const Optional< ConstTensor > &  biases, const char *  name = nullptr  )

overridevirtual

Adds a 2D convolution layer to the network.

Parameters

convolution2dDescriptor	- Description of the 2D convolution layer.
weights	- Tensor for the weights data.
biases	- Optional tensor for the bias data. If specified, must match the output tensor shape.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Implements INetwork.

Definition at line 1051 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), and Network::GetGuid().

{
    return AddConvolution2dLayerImpl(convolution2dDescriptor, weights, biases, name);
}

AddConvolution2dLayer() [2/3]

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

overridevirtual

Implements INetwork.

Definition at line 1059 of file Network.cpp.

{
    Optional<ConstTensor> biases;
    return AddConvolution2dLayerImpl(convolution2dDescriptor, weights, biases, name);
}

AddConvolution2dLayer() [3/3]

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

overridevirtual

Implements INetwork.

Definition at line 1067 of file Network.cpp.

References OptionalBase::has_value(), DepthwiseConvolution2dDescriptor::m_BiasEnabled, DepthwiseConvolution2dLayer::m_Weight, and OptionalReferenceSwitch< std::is_reference< T >::value, T >::value().

{
    Optional<ConstTensor> optionalBiases(biases);
    return AddConvolution2dLayerImpl(convolution2dDescriptor, weights, optionalBiases, name);
}

AddDepthToSpaceLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1099 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<DepthToSpaceLayer>(depthToSpaceDescriptor, name);
}

AddDepthwiseConvolution2dLayer() [1/3]

IConnectableLayer * AddDepthwiseConvolution2dLayer ( const DepthwiseConvolution2dDescriptor &  convolution2dDescriptor, const ConstTensor &  weights, const Optional< ConstTensor > &  biases, const char *  name = nullptr  )

overridevirtual

Adds a 2D depthwise convolution layer to the network.

Parameters

convolution2dDescriptor	- Description of the 2D depthwise convolution layer.
weights	- Tensor for the weights. Expected format: [channelMultiplier, inputChannels, height, width].
biases	Optional tensor for the bias data. If specified, must match the output tensor shape.
name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Implements INetwork.

Definition at line 1105 of file Network.cpp.

Referenced by armnn::BOOST_AUTO_TEST_CASE(), and Network::GetGuid().

{
    return AddDepthwiseConvolution2dLayerImpl(convolution2dDescriptor, weights, biases, name);
}

AddDepthwiseConvolution2dLayer() [2/3]

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

overridevirtual

Implements INetwork.

Definition at line 1114 of file Network.cpp.

{
    Optional<ConstTensor> biases;
    return AddDepthwiseConvolution2dLayerImpl(convolution2dDescriptor, weights, biases, name);
}

AddDepthwiseConvolution2dLayer() [3/3]

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

overridevirtual

Implements INetwork.

Definition at line 1123 of file Network.cpp.

{
    Optional<ConstTensor> optionalBiases(biases);
    return AddDepthwiseConvolution2dLayerImpl(convolution2dDescriptor, weights, optionalBiases, name);
}

AddDequantizeLayer()

IConnectableLayer * AddDequantizeLayer ( const char *  name = nullptr )

overridevirtual

Adds a Dequantize layer to the network.

Returns

- Interface for configuring the layer.

Implements INetwork.

Definition at line 1461 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<DequantizeLayer>(name);
}

AddDetectionPostProcessLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1133 of file Network.cpp.

References anchors(), and DetectionPostProcessLayer::m_Anchors.

Referenced by Network::GetGuid().

{
    const auto layer = m_Graph->AddLayer<DetectionPostProcessLayer>(descriptor, name);

    layer->m_Anchors = std::make_unique<ScopedCpuTensorHandle>(anchors);

    return layer;
}

AddDivisionLayer()

IConnectableLayer * AddDivisionLayer ( const char *  name = nullptr )

overridevirtual

Adds a division layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Implements INetwork.

Definition at line 1436 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<DivisionLayer>(name);
}

AddElementwiseUnaryLayer()

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

overridevirtual

Add an ElementwiseUnary layer to the network.

Parameters

name	- Optional name for the layer.
desc	- Descriptor for the elementwiseUnary operation. @ return - Interface for configuring the layer.

Implements INetwork.

Definition at line 970 of file Network.cpp.

References OptionalBase::has_value(), FullyConnectedDescriptor::m_BiasEnabled, FullyConnectedLayer::m_Weight, and OptionalReferenceSwitch< std::is_reference< T >::value, T >::value().

Referenced by Network::AddAbsLayer(), Network::AddRsqrtLayer(), and Network::GetGuid().

{
    return m_Graph->AddLayer<ElementwiseUnaryLayer>(elementwiseUnaryDescriptor, name);
}

AddEqualLayer()

IConnectableLayer * AddEqualLayer ( const char *  name = nullptr )

overridevirtual

Add a Equal layer to the network.

Parameters

name	- Optional name for the layer. @ return - Interface for configuring the layer.

Implements INetwork.

Definition at line 1477 of file Network.cpp.

References Network::AddComparisonLayer(), and armnn::Equal.

Referenced by Network::GetGuid().

{
    return AddComparisonLayer(ComparisonDescriptor(ComparisonOperation::Equal), name);
}

AddFloorLayer()

IConnectableLayer * AddFloorLayer ( const char *  name = nullptr )

overridevirtual

Adds a floor layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Implements INetwork.

Definition at line 1307 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), and Network::GetGuid().

{
    return m_Graph->AddLayer<FloorLayer>(name);
}

AddFullyConnectedLayer() [1/3]

IConnectableLayer * AddFullyConnectedLayer ( const FullyConnectedDescriptor &  fullyConnectedDescriptor, const ConstTensor &  weights, const Optional< ConstTensor > &  biases, const char *  name = nullptr  )

overridevirtual

Adds a fully connected layer to the network.

Parameters

fullyConnectedDescriptor	- Description of the fully connected 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.

Implements INetwork.

Definition at line 998 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), and Network::GetGuid().

{
    return AddFullyConnectedLayerImpl(fullyConnectedDescriptor, weights, biases, name);
}

AddFullyConnectedLayer() [2/3]

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

overridevirtual

Implements INetwork.

Definition at line 1006 of file Network.cpp.

{
    Optional<ConstTensor> biases;
    return AddFullyConnectedLayerImpl(fullyConnectedDescriptor, weights, biases, name);
}

AddFullyConnectedLayer() [3/3]

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

overridevirtual

Implements INetwork.

Definition at line 1014 of file Network.cpp.

{
    Optional<ConstTensor> optionalBiases(biases);
    return AddFullyConnectedLayerImpl(fullyConnectedDescriptor, weights, optionalBiases, name);
}

AddGatherLayer()

IConnectableLayer * AddGatherLayer ( const char *  name = nullptr )

overridevirtual

Add Gather layer to the network.

Parameters

name	- Optional name for the layer. @ return - Interface for configuring the layer.

Implements INetwork.

Definition at line 1487 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<GatherLayer>(name);
}

AddGreaterLayer()

IConnectableLayer * AddGreaterLayer ( const char *  name = nullptr )

overridevirtual

Add a Greater layer to the network.

Parameters

name	- Optional name for the layer. @ return - Interface for configuring the layer.

Implements INetwork.

Definition at line 1472 of file Network.cpp.

References Network::AddComparisonLayer(), and armnn::Greater.

Referenced by Network::GetGuid().

{
    return AddComparisonLayer(ComparisonDescriptor(ComparisonOperation::Greater), name);
}

AddInputLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 953 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), and Network::GetGuid().

{
    return m_Graph->AddLayer<InputLayer>(id, name);
}

AddInstanceNormalizationLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1262 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<InstanceNormalizationLayer>(desc, name);
}

AddL2NormalizationLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1268 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<L2NormalizationLayer>(desc, name);
}

AddLogSoftmaxLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1274 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<LogSoftmaxLayer>(desc, name);
}

AddLstmLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1312 of file Network.cpp.

References LstmLayer::m_BasicParameters, LstmInputParams::m_CellBias, LstmInputParams::m_CellLayerNormWeights, LstmInputParams::m_CellToForgetWeights, LstmInputParams::m_CellToInputWeights, LstmInputParams::m_CellToOutputWeights, LstmInputParams::m_ForgetGateBias, LstmInputParams::m_ForgetLayerNormWeights, LstmInputParams::m_InputGateBias, LstmInputParams::m_InputLayerNormWeights, LstmInputParams::m_InputToCellWeights, LstmInputParams::m_InputToForgetWeights, LstmBasicParameters::m_InputToForgetWeights, LstmInputParams::m_InputToInputWeights, LstmInputParams::m_InputToOutputWeights, LstmInputParams::m_OutputGateBias, LstmInputParams::m_OutputLayerNormWeights, LstmInputParams::m_ProjectionBias, LstmInputParams::m_ProjectionWeights, LstmInputParams::m_RecurrentToCellWeights, LstmInputParams::m_RecurrentToForgetWeights, LstmInputParams::m_RecurrentToInputWeights, and LstmInputParams::m_RecurrentToOutputWeights.

Referenced by armnn::BOOST_AUTO_TEST_CASE(), and Network::GetGuid().

{
    const auto layer = m_Graph->AddLayer<LstmLayer>(descriptor, name);

    //Lstm Basic Parameters
    layer->m_BasicParameters.m_InputToForgetWeights =
        std::make_unique<ScopedCpuTensorHandle>(*(params.m_InputToForgetWeights));
    layer->m_BasicParameters.m_InputToCellWeights =
        std::make_unique<ScopedCpuTensorHandle>(*(params.m_InputToCellWeights));
    layer->m_BasicParameters.m_InputToOutputWeights =
        std::make_unique<ScopedCpuTensorHandle>(*(params.m_InputToOutputWeights));
    layer->m_BasicParameters.m_RecurrentToForgetWeights =
        std::make_unique<ScopedCpuTensorHandle>(*(params.m_RecurrentToForgetWeights));
    layer->m_BasicParameters.m_RecurrentToCellWeights =
        std::make_unique<ScopedCpuTensorHandle>(*(params.m_RecurrentToCellWeights));
    layer->m_BasicParameters.m_RecurrentToOutputWeights =
        std::make_unique<ScopedCpuTensorHandle>(*(params.m_RecurrentToOutputWeights));
    layer->m_BasicParameters.m_ForgetGateBias =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_ForgetGateBias));
    layer->m_BasicParameters.m_CellBias =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_CellBias));
    layer->m_BasicParameters.m_OutputGateBias =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_OutputGateBias));

    //Lstm Cifg parameters
    if(!descriptor.m_CifgEnabled)
    {
        if(params.m_InputToInputWeights == nullptr)
        {
            throw InvalidArgumentException("AddLstmLayer: Input To Input Weights cannot be NULL");
        }
        if(params.m_RecurrentToInputWeights == nullptr)
        {
            throw InvalidArgumentException(
                    "AddLstmLayer: Recurrent To Input Weights cannot be NULL");
        }
        if(params.m_InputGateBias == nullptr)
        {
            throw InvalidArgumentException("AddLstmLayer: Input Gate Bias cannot be NULL");
        }
        layer->m_CifgParameters.m_InputToInputWeights =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_InputToInputWeights));
        layer->m_CifgParameters.m_RecurrentToInputWeights =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_RecurrentToInputWeights));
        // In the VTS tests, cell-to-input weights may be null, even if the other CIFG params are not.
        if(params.m_CellToInputWeights != nullptr)
        {
            layer->m_CifgParameters.m_CellToInputWeights =
                    std::make_unique<ScopedCpuTensorHandle>(*(params.m_CellToInputWeights));
        }
        layer->m_CifgParameters.m_InputGateBias =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_InputGateBias));
    }

    //Lstm projection parameters
    if(descriptor.m_ProjectionEnabled)
    {
        if(params.m_ProjectionWeights == nullptr)
        {
            throw InvalidArgumentException("AddLstmLayer: Projection Weights cannot be NULL");
        }
        layer->m_ProjectionParameters.m_ProjectionWeights =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_ProjectionWeights));
        if(params.m_ProjectionBias != nullptr)
        {
            layer->m_ProjectionParameters.m_ProjectionBias =
                std::make_unique<ScopedCpuTensorHandle>(*(params.m_ProjectionBias));
        }
    }

    //Lstm Peephole params
    if(descriptor.m_PeepholeEnabled)
    {
        if(params.m_CellToForgetWeights == nullptr)
        {
            throw InvalidArgumentException("AddLstmLayer: Cell To Forget Weights cannot be NULL");
        }
        if(params.m_CellToOutputWeights == nullptr)
        {
            throw InvalidArgumentException("AddLstmLayer: Cell To Output Weights cannot be NULL");
        }
        layer->m_PeepholeParameters.m_CellToForgetWeights =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_CellToForgetWeights));
        layer->m_PeepholeParameters.m_CellToOutputWeights =
            std::make_unique<ScopedCpuTensorHandle>(*(params.m_CellToOutputWeights));
    }

    //Lstm Layer Normalization params
    if(descriptor.m_LayerNormEnabled)
    {
        if(!descriptor.m_CifgEnabled)
        {
            if(params.m_InputLayerNormWeights == nullptr)
            {
                throw InvalidArgumentException("AddLstmLayer: Input layer normalization weights cannot be NULL");
            }
            layer->m_LayerNormParameters.m_InputLayerNormWeights =
                    std::make_unique<ScopedCpuTensorHandle>(*(params.m_InputLayerNormWeights));
        }

        if(params.m_ForgetLayerNormWeights == nullptr)
        {
            throw InvalidArgumentException("AddLstmLayer: Forget layer normalization weights cannot be NULL");
        }
        if(params.m_CellLayerNormWeights == nullptr)
        {
            throw InvalidArgumentException("AddLstmLayer: Cell layer normalization weights cannot be NULL");
        }
        if(params.m_OutputLayerNormWeights == nullptr)
        {
            throw InvalidArgumentException("AddLstmLayer: Output layer normalization weights cannot be NULL");
        }
        layer->m_LayerNormParameters.m_ForgetLayerNormWeights =
                std::make_unique<ScopedCpuTensorHandle>(*(params.m_ForgetLayerNormWeights));
        layer->m_LayerNormParameters.m_CellLayerNormWeights =
                std::make_unique<ScopedCpuTensorHandle>(*(params.m_CellLayerNormWeights));
        layer->m_LayerNormParameters.m_OutputLayerNormWeights =
                std::make_unique<ScopedCpuTensorHandle>(*(params.m_OutputLayerNormWeights));
    }
    return layer;
}

AddMaximumLayer()

IConnectableLayer * AddMaximumLayer ( const char *  name = nullptr )

overridevirtual

Add a Maximum layer to the network.

Parameters

name	- Optional name for the layer. @ return - Interface for configuring the layer.

Implements INetwork.

Definition at line 1191 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<MaximumLayer>(name);
}

AddMeanLayer()

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

overridevirtual

Add a Mean layer to the network.

Parameters

meanDescriptor	- Parameters for the mean operation.
name	- Optional name for the layer. @ return - Interface for configuring the layer.

Implements INetwork.

Definition at line 1446 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<MeanLayer>(meanDescriptor,name);
}

AddMergeLayer()

IConnectableLayer * AddMergeLayer ( const char *  name = nullptr )

overridevirtual

Adds a merge layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Implements INetwork.

Definition at line 1492 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<MergeLayer>(name);
}

AddMergerLayer()

IConnectableLayer * AddMergerLayer ( const MergerDescriptor &  mergerDescriptor, const char *  name = nullptr  )

overridevirtual

Adds a concat layer to the network.

Parameters

mergerDescriptor	- MergerDescriptor (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.

Implements INetwork.

Definition at line 1201 of file Network.cpp.

References Network::AddConcatLayer().

Referenced by Network::GetGuid().

{
    return AddConcatLayer(mergerDescriptor, name);
}

AddMinimumLayer()

IConnectableLayer * AddMinimumLayer ( const char *  name = nullptr )

overridevirtual

Add a Minimum layer to the network.

Parameters

name	- Optional name for the layer. @ return - Interface for configuring the layer.

Implements INetwork.

Definition at line 1196 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<MinimumLayer>(name);
}

AddMultiplicationLayer()

IConnectableLayer * AddMultiplicationLayer ( const char *  name = nullptr )

overridevirtual

Adds a multiplication layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Implements INetwork.

Definition at line 1217 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), and Network::GetGuid().

{
    return m_Graph->AddLayer<MultiplicationLayer>(name);
}

AddNormalizationLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1167 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), and Network::GetGuid().

{
    return m_Graph->AddLayer<NormalizationLayer>(normalizationDescriptor, name);
}

AddOutputLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1222 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), and Network::GetGuid().

{
    return m_Graph->AddLayer<OutputLayer>(id, name);
}

AddPadLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1451 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<PadLayer>(padDescriptor,name);
}

AddPermuteLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1143 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<PermuteLayer>(permuteDescriptor, name);
}

AddPooling2dLayer()

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

overridevirtual

Adds a pooling layer to the network.

Parameters

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

Returns

- Interface for configuring the layer.

Implements INetwork.

Definition at line 1149 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), and Network::GetGuid().

{
    return m_Graph->AddLayer<Pooling2dLayer>(pooling2dDescriptor, name);
}

AddPreluLayer()

IConnectableLayer * AddPreluLayer ( const char *  name = nullptr )

overridevirtual

Adds a PReLU layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Implements INetwork.

Definition at line 1502 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<PreluLayer>(name);
}

AddQuantizedLstmLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1542 of file Network.cpp.

References QuantizedLstmInputParams::GetCellBias(), QuantizedLstmInputParams::GetForgetGateBias(), QuantizedLstmInputParams::GetInputGateBias(), QuantizedLstmInputParams::GetInputToCellWeights(), QuantizedLstmInputParams::GetInputToForgetWeights(), QuantizedLstmInputParams::GetInputToInputWeights(), QuantizedLstmInputParams::GetInputToOutputWeights(), QuantizedLstmInputParams::GetOutputGateBias(), QuantizedLstmInputParams::GetRecurrentToCellWeights(), QuantizedLstmInputParams::GetRecurrentToForgetWeights(), QuantizedLstmInputParams::GetRecurrentToInputWeights(), QuantizedLstmInputParams::GetRecurrentToOutputWeights(), QuantizedLstmParameters::m_InputToInputWeights, and QuantizedLstmLayer::m_QuantizedLstmParameters.

Referenced by armnn::BOOST_AUTO_TEST_CASE(), and Network::GetGuid().

{
    const auto layer = m_Graph->AddLayer<QuantizedLstmLayer>(name);

    // InputToX weights
    layer->m_QuantizedLstmParameters.m_InputToInputWeights =
            std::make_unique<ScopedCpuTensorHandle>(params.GetInputToInputWeights());
    layer->m_QuantizedLstmParameters.m_InputToForgetWeights =
            std::make_unique<ScopedCpuTensorHandle>(params.GetInputToForgetWeights());
    layer->m_QuantizedLstmParameters.m_InputToCellWeights =
            std::make_unique<ScopedCpuTensorHandle>(params.GetInputToCellWeights());
    layer->m_QuantizedLstmParameters.m_InputToOutputWeights =
            std::make_unique<ScopedCpuTensorHandle>(params.GetInputToOutputWeights());

    // RecurrentToX weights
    layer->m_QuantizedLstmParameters.m_RecurrentToInputWeights =
            std::make_unique<ScopedCpuTensorHandle>(params.GetRecurrentToInputWeights());
    layer->m_QuantizedLstmParameters.m_RecurrentToForgetWeights =
            std::make_unique<ScopedCpuTensorHandle>(params.GetRecurrentToForgetWeights());
    layer->m_QuantizedLstmParameters.m_RecurrentToCellWeights =
            std::make_unique<ScopedCpuTensorHandle>(params.GetRecurrentToCellWeights());
    layer->m_QuantizedLstmParameters.m_RecurrentToOutputWeights =
            std::make_unique<ScopedCpuTensorHandle>(params.GetRecurrentToOutputWeights());

    // Bias
    layer->m_QuantizedLstmParameters.m_InputGateBias =
            std::make_unique<ScopedCpuTensorHandle>(params.GetInputGateBias());
    layer->m_QuantizedLstmParameters.m_ForgetGateBias =
            std::make_unique<ScopedCpuTensorHandle>(params.GetForgetGateBias());
    layer->m_QuantizedLstmParameters.m_CellBias =
            std::make_unique<ScopedCpuTensorHandle>(params.GetCellBias());
    layer->m_QuantizedLstmParameters.m_OutputGateBias =
            std::make_unique<ScopedCpuTensorHandle>(params.GetOutputGateBias());

    return layer;
}

AddQuantizeLayer()

IConnectableLayer * AddQuantizeLayer ( const char *  name = nullptr )

overridevirtual

Add a quantize layer to the network

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Implements INetwork.

Definition at line 1456 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<QuantizeLayer>(name);
}

AddReshapeLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1289 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<ReshapeLayer>(reshapeDescriptor, name);
}

AddResizeBilinearLayer()

IConnectableLayer * AddResizeBilinearLayer ( const ResizeBilinearDescriptor &  resizeDesc, const char *  name = nullptr  )

overridevirtual

Adds a resize bilinear layer to the network.

Parameters

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

Returns

- Interface for configuring the layer.

Implements INetwork.

Definition at line 1244 of file Network.cpp.

References armnn::Bilinear, ResizeBilinearDescriptor::m_DataLayout, ResizeDescriptor::m_DataLayout, ResizeDescriptor::m_Method, ResizeBilinearDescriptor::m_TargetHeight, ResizeDescriptor::m_TargetHeight, ResizeBilinearDescriptor::m_TargetWidth, and ResizeDescriptor::m_TargetWidth.

Referenced by Network::GetGuid().

{
    ResizeDescriptor resizeDescriptor;
    resizeDescriptor.m_Method       = ResizeMethod::Bilinear;
    resizeDescriptor.m_DataLayout   = descriptor.m_DataLayout;
    resizeDescriptor.m_TargetWidth  = descriptor.m_TargetWidth;
    resizeDescriptor.m_TargetHeight = descriptor.m_TargetHeight;

    return m_Graph->AddLayer<ResizeLayer>(resizeDescriptor, name);
}

AddResizeLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1256 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<ResizeLayer>(resizeDescriptor, name);
}

AddRsqrtLayer()

IConnectableLayer * AddRsqrtLayer ( const char *  name = nullptr )

overridevirtual

Add Reciprocal of square root layer to the network.

Parameters

name	- Optional name for the layer. @ return - Interface for configuring the layer.

Implements INetwork.

Definition at line 1482 of file Network.cpp.

References Network::AddElementwiseUnaryLayer(), and armnn::Rsqrt.

Referenced by Network::GetGuid().

{
    return AddElementwiseUnaryLayer(ElementwiseUnaryDescriptor(UnaryOperation::Rsqrt), name);
}

AddSliceLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1174 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<SliceLayer>(sliceDescriptor, name);
}

AddSoftmaxLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1179 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), and Network::GetGuid().

{
    return m_Graph->AddLayer<SoftmaxLayer>(softmaxDescriptor, name);
}

AddSpaceToBatchNdLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1295 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<SpaceToBatchNdLayer>(spaceToBatchNdDescriptor, name);
}

AddSpaceToDepthLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1301 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<SpaceToDepthLayer>(spaceToDepthDescriptor, name);
}

AddSplitterLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1185 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), and Network::GetGuid().

{
    return m_Graph->AddLayer<SplitterLayer>(splitterDescriptor, name);
}

AddStackLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1529 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<StackLayer>(stackDescriptor, name);
}

AddStandInLayer()

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

overridevirtual

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 for the StandIn layer.

Returns

- Interface for configuring the layer.

Implements INetwork.

Definition at line 1536 of file Network.cpp.

Referenced by BOOST_AUTO_TEST_CASE(), and Network::GetGuid().

{
    return m_Graph->AddLayer<StandInLayer>(desc, name);
}

AddStridedSliceLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1466 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<StridedSliceLayer>(stridedSliceDescriptor, name);
}

AddSubtractionLayer()

IConnectableLayer * AddSubtractionLayer ( const char *  name = nullptr )

overridevirtual

Adds a subtraction layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Implements INetwork.

Definition at line 1441 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<SubtractionLayer>(name);
}

AddSwitchLayer()

IConnectableLayer * AddSwitchLayer ( const char *  name = nullptr )

overridevirtual

Adds a switch layer to the network.

Parameters

name	- Optional name for the layer.

Returns

- Interface for configuring the layer.

Implements INetwork.

Definition at line 1497 of file Network.cpp.

Referenced by Network::GetGuid().

{
    return m_Graph->AddLayer<SwitchLayer>(name);
}

AddTransposeConvolution2dLayer()

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

overridevirtual

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.

Implements INetwork.

Definition at line 1507 of file Network.cpp.

References OptionalBase::has_value(), TransposeConvolution2dDescriptor::m_BiasEnabled, TransposeConvolution2dLayer::m_Weight, and OptionalReferenceSwitch< std::is_reference< T >::value, T >::value().

Referenced by Network::GetGuid().

{
    if (descriptor.m_BiasEnabled && !biases.has_value())
    {
        throw InvalidArgumentException("AddTransposeConvolution2dLayer: Biases cannot be empty");
    }

    const auto layer = m_Graph->AddLayer<TransposeConvolution2dLayer>(descriptor, name);

    layer->m_Weight = std::make_unique<ScopedCpuTensorHandle>(weights);

    if (descriptor.m_BiasEnabled)
    {
        layer->m_Bias = std::make_unique<ScopedCpuTensorHandle>(biases.value());
    }

    return layer;
}

GetGraph()

const Graph& GetGraph ( ) const

inline

Definition at line 33 of file Network.hpp.

References Network::PrintGraph().

Referenced by Network::Accept(), BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), armnn::GetInputTensorInfo(), and armnn::Optimize().

{ return *m_Graph; }

GetGuid()

profiling::ProfilingGuid GetGuid ( ) const

inlinefinalvirtual

Implements INetwork.

Definition at line 37 of file Network.hpp.

References Network::Accept(), Network::AddAbsLayer(), Network::AddActivationLayer(), Network::AddAdditionLayer(), Network::AddArgMinMaxLayer(), Network::AddBatchNormalizationLayer(), Network::AddBatchToSpaceNdLayer(), Network::AddComparisonLayer(), Network::AddConcatLayer(), Network::AddConstantLayer(), Network::AddConvolution2dLayer(), Network::AddDepthToSpaceLayer(), Network::AddDepthwiseConvolution2dLayer(), Network::AddDequantizeLayer(), Network::AddDetectionPostProcessLayer(), Network::AddDivisionLayer(), Network::AddElementwiseUnaryLayer(), Network::AddEqualLayer(), Network::AddFloorLayer(), Network::AddFullyConnectedLayer(), Network::AddGatherLayer(), Network::AddGreaterLayer(), Network::AddInputLayer(), Network::AddInstanceNormalizationLayer(), Network::AddL2NormalizationLayer(), Network::AddLogSoftmaxLayer(), Network::AddLstmLayer(), Network::AddMaximumLayer(), Network::AddMeanLayer(), Network::AddMergeLayer(), Network::AddMergerLayer(), Network::AddMinimumLayer(), Network::AddMultiplicationLayer(), Network::AddNormalizationLayer(), Network::AddOutputLayer(), Network::AddPadLayer(), Network::AddPermuteLayer(), Network::AddPooling2dLayer(), Network::AddPreluLayer(), Network::AddQuantizedLstmLayer(), Network::AddQuantizeLayer(), Network::AddReshapeLayer(), Network::AddResizeBilinearLayer(), Network::AddResizeLayer(), Network::AddRsqrtLayer(), Network::AddSliceLayer(), Network::AddSoftmaxLayer(), Network::AddSpaceToBatchNdLayer(), Network::AddSpaceToDepthLayer(), Network::AddSplitterLayer(), Network::AddStackLayer(), Network::AddStandInLayer(), Network::AddStridedSliceLayer(), Network::AddSubtractionLayer(), Network::AddSwitchLayer(), Network::AddTransposeConvolution2dLayer(), anchors(), and ARMNN_DEPRECATED_MSG.

{ return m_Guid; };

PrintGraph()

Status PrintGraph ( )

overridevirtual

Implements INetwork.

Definition at line 947 of file Network.cpp.

References armnn::Success.

Referenced by BOOST_AUTO_TEST_CASE(), and Network::GetGraph().

{
    m_Graph->Print();
    return Status::Success;
}

---

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

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