20.05
|
Private implementation of INetwork. More...
#include <Network.hpp>
Public Member Functions | |
Network () | |
~Network () | |
const Graph & | GetGraph () const |
Status | PrintGraph () override |
IConnectableLayer * | AddInputLayer (LayerBindingId id, const char *name=nullptr) override |
Adds an input layer to the network. More... | |
IConnectableLayer * | AddArgMinMaxLayer (const ArgMinMaxDescriptor &desc, const char *name=nullptr) override |
Adds an ArgMinMax layer to the network. More... | |
IConnectableLayer * | AddBatchToSpaceNdLayer (const BatchToSpaceNdDescriptor &batchToSpaceNdDescriptor, const char *name=nullptr) override |
Adds a batch to space ND layer to the network. More... | |
IConnectableLayer * | AddComparisonLayer (const ComparisonDescriptor &comparisonDescriptor, const char *name=nullptr) override |
Add a Comparison layer to the network. More... | |
IConnectableLayer * | AddConcatLayer (const ConcatDescriptor &concatDescriptor, const char *name=nullptr) override |
Adds a concatenation layer to the network. More... | |
IConnectableLayer * | AddConvolution2dLayer (const Convolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr) override |
Adds a 2D convolution layer to the network. More... | |
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 |
Adds a depth to space layer to the network. More... | |
IConnectableLayer * | AddDepthwiseConvolution2dLayer (const DepthwiseConvolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr) override |
Adds a 2D depthwise convolution layer to the network. More... | |
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 |
Adds a Dequantize layer to the network. More... | |
IConnectableLayer * | AddDetectionPostProcessLayer (const DetectionPostProcessDescriptor &descriptor, const ConstTensor &anchors, const char *name=nullptr) override |
Adds a Detection PostProcess layer to the network. More... | |
IConnectableLayer * | AddElementwiseUnaryLayer (const ElementwiseUnaryDescriptor &elementwiseUnaryDescriptor, const char *name=nullptr) override |
Add an ElementwiseUnary layer to the network. More... | |
IConnectableLayer * | AddFullyConnectedLayer (const FullyConnectedDescriptor &fullyConnectedDescriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr) override |
Adds a fully connected layer to the network. More... | |
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 |
Add Gather layer to the network. More... | |
IConnectableLayer * | AddPermuteLayer (const PermuteDescriptor &permuteDescriptor, const char *name=nullptr) override |
Adds a permute layer to the network. More... | |
IConnectableLayer * | AddPooling2dLayer (const Pooling2dDescriptor &pooling2dDescriptor, const char *name=nullptr) override |
Adds a pooling layer to the network. More... | |
IConnectableLayer * | AddActivationLayer (const ActivationDescriptor &activationDescriptor, const char *name=nullptr) override |
Adds an activation layer to the network. More... | |
IConnectableLayer * | AddNormalizationLayer (const NormalizationDescriptor &normalizationDescriptor, const char *name=nullptr) override |
Adds a normalization layer to the network. More... | |
IConnectableLayer * | AddSliceLayer (const SliceDescriptor &sliceDescriptor, const char *name=nullptr) override |
Adds a slice layer to the network. More... | |
IConnectableLayer * | AddSoftmaxLayer (const SoftmaxDescriptor &softmaxDescriptor, const char *name=nullptr) override |
Adds a softmax layer to the network. More... | |
IConnectableLayer * | AddSplitterLayer (const ViewsDescriptor &splitterDescriptor, const char *name=nullptr) override |
Adds a splitter layer to the network. More... | |
IConnectableLayer * | AddMergerLayer (const MergerDescriptor &mergerDescriptor, const char *name=nullptr) override |
Adds a concat layer to the network. More... | |
IConnectableLayer * | AddAbsLayer (const char *name=nullptr) override |
Add absolute layer to the network. More... | |
IConnectableLayer * | AddAdditionLayer (const char *name=nullptr) override |
Adds an addition layer to the network. More... | |
IConnectableLayer * | AddMultiplicationLayer (const char *name=nullptr) override |
Adds a multiplication layer to the network. More... | |
IConnectableLayer * | AddBatchNormalizationLayer (const BatchNormalizationDescriptor &desc, const ConstTensor &mean, const ConstTensor &variance, const ConstTensor &beta, const ConstTensor &gamma, const char *name=nullptr) override |
Adds a batch normalization layer to the network. More... | |
IConnectableLayer * | AddResizeBilinearLayer (const ResizeBilinearDescriptor &resizeDesc, const char *name=nullptr) override |
Adds a resize bilinear layer to the network. More... | |
IConnectableLayer * | AddResizeLayer (const ResizeDescriptor &resizeDescriptor, const char *name=nullptr) override |
Adds a resize layer to the network. More... | |
IConnectableLayer * | AddInstanceNormalizationLayer (const InstanceNormalizationDescriptor &desc, const char *name=nullptr) override |
Adds an instance normalization layer to the network. More... | |
IConnectableLayer * | AddL2NormalizationLayer (const L2NormalizationDescriptor &desc, const char *name=nullptr) override |
Adds an L2 normalization layer to the network. More... | |
IConnectableLayer * | AddLogSoftmaxLayer (const LogSoftmaxDescriptor &logSoftmaxDescriptor, const char *name=nullptr) override |
Adds a log softmax layer to the network. More... | |
IConnectableLayer * | AddConstantLayer (const ConstTensor &input, const char *name=nullptr) override |
Adds a layer with no inputs and a single output, which always corresponds to the passed in constant tensor. More... | |
IConnectableLayer * | AddReshapeLayer (const ReshapeDescriptor &reshapeDescriptor, const char *name=nullptr) override |
Adds a reshape layer to the network. More... | |
IConnectableLayer * | AddSpaceToBatchNdLayer (const SpaceToBatchNdDescriptor &spaceToBatchNdDescriptor, const char *name=nullptr) override |
Adds a space to batch layer to the network. More... | |
IConnectableLayer * | AddSpaceToDepthLayer (const SpaceToDepthDescriptor &spaceToDepthDescriptor, const char *name=nullptr) override |
Adds a space to depth layer to the network. More... | |
IConnectableLayer * | AddFloorLayer (const char *name=nullptr) override |
Adds a floor layer to the network. More... | |
IConnectableLayer * | AddOutputLayer (LayerBindingId id, const char *name=nullptr) override |
Adds an output layer to the network. More... | |
IConnectableLayer * | AddLstmLayer (const LstmDescriptor &descriptor, const LstmInputParams ¶ms, const char *name=nullptr) override |
Add a Lstm layer to the network. More... | |
IConnectableLayer * | AddDivisionLayer (const char *name=nullptr) override |
Adds a division layer to the network. More... | |
IConnectableLayer * | AddSubtractionLayer (const char *name=nullptr) override |
Adds a subtraction layer to the network. More... | |
IConnectableLayer * | AddMaximumLayer (const char *name=nullptr) override |
Add a Maximum layer to the network. More... | |
IConnectableLayer * | AddMeanLayer (const MeanDescriptor &meanDescriptor, const char *name=nullptr) override |
Add a Mean layer to the network. More... | |
IConnectableLayer * | AddPadLayer (const PadDescriptor &padDescriptor, const char *name=nullptr) override |
Adds a fully pad layer to the network. More... | |
IConnectableLayer * | AddQuantizeLayer (const char *name=nullptr) override |
Add a quantize layer to the network. More... | |
IConnectableLayer * | AddStridedSliceLayer (const StridedSliceDescriptor &stridedSliceDescriptor, const char *name=nullptr) override |
Adds a strided slice layer to the network. More... | |
IConnectableLayer * | AddMinimumLayer (const char *name=nullptr) override |
Add a Minimum layer to the network. More... | |
IConnectableLayer * | AddGreaterLayer (const char *name=nullptr) override |
Add a Greater layer to the network. More... | |
IConnectableLayer * | AddEqualLayer (const char *name=nullptr) override |
Add a Equal layer to the network. More... | |
IConnectableLayer * | AddRsqrtLayer (const char *name=nullptr) override |
Add Reciprocal of square root layer to the network. More... | |
IConnectableLayer * | AddMergeLayer (const char *name=nullptr) override |
Adds a merge layer to the network. More... | |
IConnectableLayer * | AddSwitchLayer (const char *name=nullptr) override |
Adds a switch layer to the network. More... | |
IConnectableLayer * | AddPreluLayer (const char *name=nullptr) override |
Adds a PReLU layer to the network. More... | |
IConnectableLayer * | AddTransposeConvolution2dLayer (const TransposeConvolution2dDescriptor &descriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr) override |
Adds a 2D transpose convolution layer to the network. More... | |
IConnectableLayer * | AddTransposeLayer (const TransposeDescriptor &transposeDescriptor, const char *name=nullptr) override |
Adds a transpose layer to the network. More... | |
IConnectableLayer * | AddStackLayer (const StackDescriptor &stackDescriptor, const char *name=nullptr) override |
Adds a stack layer to the network. More... | |
IConnectableLayer * | AddStandInLayer (const StandInDescriptor &descriptor, const char *name=nullptr) override |
Add a stand-in layer for a type unknown to the Arm NN framework. More... | |
IConnectableLayer * | AddQLstmLayer (const QLstmDescriptor &descriptor, const LstmInputParams ¶ms, const char *name=nullptr) override |
Add a QLstm layer to the network. More... | |
IConnectableLayer * | AddQuantizedLstmLayer (const QuantizedLstmInputParams ¶ms, const char *name=nullptr) override |
Add a QuantizedLstm layer to the network. More... | |
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 () | |
Private implementation of INetwork.
Definition at line 28 of file Network.hpp.
Network | ( | ) |
Definition at line 1151 of file Network.cpp.
~Network | ( | ) |
Definition at line 1156 of file Network.cpp.
|
overridevirtual |
Implements INetwork.
Definition at line 1958 of file Network.cpp.
References Network::GetGraph().
Referenced by Network::GetGraph().
|
overridevirtual |
Add absolute layer to the network.
name | - Optional name for the layer. @ return - Interface for configuring the layer. |
Implements INetwork.
Definition at line 1420 of file Network.cpp.
References armnn::Abs, and Network::AddElementwiseUnaryLayer().
Referenced by Network::GetGraph().
|
overridevirtual |
Adds an activation layer to the network.
activationDescriptor | - ActivationDescriptor to configure the activation. |
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1368 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), and Network::GetGraph().
|
overridevirtual |
Adds an addition layer to the network.
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1425 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), and Network::GetGraph().
|
overridevirtual |
Adds an ArgMinMax layer to the network.
desc | - Parameters for the L2 normalization operation. |
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1374 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Adds a batch normalization layer to the network.
mean | - Pre-calculated mean for each channel. |
variance | - Pre-calculated variance for each channel. |
beta | - Per-channel additive factor. |
gamma | - Per-channel multiplicative factor. |
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1440 of file Network.cpp.
References BatchNormalizationLayer::m_Mean.
Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), and Network::GetGraph().
|
overridevirtual |
Adds a batch to space ND layer to the network.
batchToSpaceNdDescriptor | - Description of the layer. |
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1171 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Add a Comparison layer to the network.
name | - Optional name for the layer. |
desc | - Descriptor for the comparison operation. @ return - Interface for configuring the layer. |
Implements INetwork.
Definition at line 1177 of file Network.cpp.
Referenced by Network::AddEqualLayer(), Network::AddGreaterLayer(), and Network::GetGraph().
|
overridevirtual |
Adds a concatenation layer to the network.
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. |
Implements INetwork.
Definition at line 1236 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::GetGraph().
|
overridevirtual |
Adds a layer with no inputs and a single output, which always corresponds to the passed in constant tensor.
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. |
Implements INetwork.
Definition at line 1493 of file Network.cpp.
References ConstantLayer::m_LayerOutput.
Referenced by armnn::BOOST_AUTO_TEST_CASE(), and Network::GetGraph().
|
overridevirtual |
Adds a 2D convolution layer to the network.
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. |
Implements INetwork.
Definition at line 1264 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), and Network::GetGraph().
|
overridevirtual |
Implements INetwork.
Definition at line 1272 of file Network.cpp.
|
overridevirtual |
Implements INetwork.
Definition at line 1280 of file Network.cpp.
References OptionalBase::has_value(), DepthwiseConvolution2dDescriptor::m_BiasEnabled, DepthwiseConvolution2dLayer::m_Weight, and OptionalReferenceSwitch< std::is_reference< T >::value, T >::value().
|
overridevirtual |
Adds a depth to space layer to the network.
depthToSpaceDescriptor | - Parameters for the depth to space operation. |
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1312 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Adds a 2D depthwise convolution layer to the network.
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. |
Implements INetwork.
Definition at line 1318 of file Network.cpp.
Referenced by armnn::BOOST_AUTO_TEST_CASE(), and Network::GetGraph().
|
overridevirtual |
Implements INetwork.
Definition at line 1327 of file Network.cpp.
|
overridevirtual |
Implements INetwork.
Definition at line 1336 of file Network.cpp.
|
overridevirtual |
Adds a Dequantize layer to the network.
Implements INetwork.
Definition at line 1691 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Adds a Detection PostProcess layer to the network.
descriptor | - Description of the Detection PostProcess layer. |
anchors | - Tensor for anchors. |
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1346 of file Network.cpp.
References anchors(), and DetectionPostProcessLayer::m_Anchors.
Referenced by Network::GetGraph().
|
overridevirtual |
Adds a division layer to the network.
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1666 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Add an ElementwiseUnary layer to the network.
name | - Optional name for the layer. |
desc | - Descriptor for the elementwiseUnary operation. @ return - Interface for configuring the layer. |
Implements INetwork.
Definition at line 1183 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::GetGraph().
|
overridevirtual |
Add a Equal layer to the network.
name | - Optional name for the layer. @ return - Interface for configuring the layer. |
Implements INetwork.
Definition at line 1707 of file Network.cpp.
References Network::AddComparisonLayer(), and armnn::Equal.
Referenced by Network::GetGraph().
|
overridevirtual |
Adds a floor layer to the network.
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1520 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), and Network::GetGraph().
|
overridevirtual |
Adds a fully connected layer to the network.
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. |
Implements INetwork.
Definition at line 1211 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), and Network::GetGraph().
|
overridevirtual |
Implements INetwork.
Definition at line 1219 of file Network.cpp.
|
overridevirtual |
Implements INetwork.
Definition at line 1227 of file Network.cpp.
|
overridevirtual |
Add Gather layer to the network.
name | - Optional name for the layer. @ return - Interface for configuring the layer. |
Implements INetwork.
Definition at line 1717 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Add a Greater layer to the network.
name | - Optional name for the layer. @ return - Interface for configuring the layer. |
Implements INetwork.
Definition at line 1702 of file Network.cpp.
References Network::AddComparisonLayer(), and armnn::Greater.
Referenced by Network::GetGraph().
|
overridevirtual |
Adds an input layer to the network.
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. |
Implements INetwork.
Definition at line 1166 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), and Network::GetGraph().
|
overridevirtual |
Adds an instance normalization layer to the network.
desc | - Parameters for the instance normalization operation. |
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1475 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Adds an L2 normalization layer to the network.
Normalization is performed along dimension 1, but requires a 4d input.
desc | - Parameters for the L2 normalization operation. |
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1481 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Adds a log softmax layer to the network.
logSoftmaxDescriptor | - LogSoftmaxDescriptor to configure the log softmax. |
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1487 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Add a Lstm layer to the network.
descriptor | - Parameters for the Lstm operation |
params | - Weights and biases for the LSTM cell |
name | - Optional name for the layer |
Implements INetwork.
Definition at line 1525 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::GetGraph().
|
overridevirtual |
Add a Maximum layer to the network.
name | - Optional name for the layer. @ return - Interface for configuring the layer. |
Implements INetwork.
Definition at line 1404 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Add a Mean layer to the network.
meanDescriptor | - Parameters for the mean operation. |
name | - Optional name for the layer. @ return - Interface for configuring the layer. |
Implements INetwork.
Definition at line 1676 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Adds a merge layer to the network.
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1722 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Adds a concat layer to the network.
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. |
Implements INetwork.
Definition at line 1414 of file Network.cpp.
References Network::AddConcatLayer().
Referenced by Network::GetGraph().
|
overridevirtual |
Add a Minimum layer to the network.
name | - Optional name for the layer. @ return - Interface for configuring the layer. |
Implements INetwork.
Definition at line 1409 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Adds a multiplication layer to the network.
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1430 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), and Network::GetGraph().
|
overridevirtual |
Adds a normalization layer to the network.
normalizationDescriptor | - NormalizationDescriptor to configure the normalization. |
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1380 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), and Network::GetGraph().
|
overridevirtual |
Adds an output layer to the network.
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. |
Implements INetwork.
Definition at line 1435 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), and Network::GetGraph().
|
overridevirtual |
Adds a fully pad layer to the network.
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. |
Implements INetwork.
Definition at line 1681 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Adds a permute layer to the network.
permuteDescriptor | - PermuteDescriptor to configure the permute. |
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1356 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Adds a pooling layer to the network.
pooling2dDescriptor | - Pooling2dDescriptor to configure the pooling. |
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1362 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), and Network::GetGraph().
|
overridevirtual |
Adds a PReLU layer to the network.
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1732 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Add a QLstm layer to the network.
descriptor | - Parameters for the QLstm operation |
params | - Weights and biases for the layer |
name | - Optional name for the layer |
Implements INetwork.
Definition at line 1816 of file Network.cpp.
References QLstmLayer::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, QLstmBasicParameters::m_InputToForgetWeights, LstmInputParams::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::GetGraph().
|
overridevirtual |
Add a QuantizedLstm layer to the network.
params | - The weights and biases for the Quantized LSTM cell |
name | - Optional name for the layer |
Implements INetwork.
Definition at line 1778 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::GetGraph().
|
overridevirtual |
Add a quantize layer to the network.
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1686 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Adds a reshape layer to the network.
reshapeDescriptor | - Parameters for the reshape operation. |
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1502 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Adds a resize bilinear layer to the network.
resizeDesc | - Parameters for the resize operation. |
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1457 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::GetGraph().
|
overridevirtual |
Adds a resize layer to the network.
resizeDescriptor | - Parameters for the resize operation. |
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1469 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Add Reciprocal of square root layer to the network.
name | - Optional name for the layer. @ return - Interface for configuring the layer. |
Implements INetwork.
Definition at line 1712 of file Network.cpp.
References Network::AddElementwiseUnaryLayer(), and armnn::Rsqrt.
Referenced by Network::GetGraph().
|
overridevirtual |
Adds a slice layer to the network.
sliceDescriptor | - SliceDescriptor to configure the slice operation. |
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1387 of file Network.cpp.
Referenced by Network::GetGraph().
|
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
softmaxDescriptor | - SoftmaxDescriptor to configure the softmax. |
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1392 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), and Network::GetGraph().
|
overridevirtual |
Adds a space to batch layer to the network.
spaceToBatchNdDescriptor | - Parameters for the space to batch operation. |
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1508 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Adds a space to depth layer to the network.
spaceToDepthDescriptor | - Parameters for the space to depth operation. |
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1514 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Adds a splitter layer to the network.
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. |
Implements INetwork.
Definition at line 1398 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), and Network::GetGraph().
|
overridevirtual |
Adds a stack layer to the network.
descriptor | - Description of the stack layer. |
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1765 of file Network.cpp.
Referenced by Network::GetGraph().
|
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.
Implements INetwork.
Definition at line 1772 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), and Network::GetGraph().
|
overridevirtual |
Adds a strided slice layer to the network.
StridedSliceDescriptor | - Parameters for the strided slice operation. |
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1696 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Adds a subtraction layer to the network.
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1671 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Adds a switch layer to the network.
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1727 of file Network.cpp.
Referenced by Network::GetGraph().
|
overridevirtual |
Adds a 2D transpose convolution layer to the network.
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. |
Implements INetwork.
Definition at line 1737 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::GetGraph().
|
overridevirtual |
Adds a transpose layer to the network.
transposeDescriptor | - TransposeDescriptor to configure the transpose. |
name | - Optional name for the layer. |
Implements INetwork.
Definition at line 1759 of file Network.cpp.
Referenced by Network::GetGraph().
|
inline |
Definition at line 34 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::AddQLstmLayer(), 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(), Network::AddTransposeLayer(), anchors(), ARMNN_DEPRECATED_MSG, and Network::PrintGraph().
Referenced by Network::Accept(), BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), armnn::GetInputTensorInfo(), and armnn::Optimize().
|
overridevirtual |
Implements INetwork.
Definition at line 1160 of file Network.cpp.
References armnn::Success.
Referenced by BOOST_AUTO_TEST_CASE(), and Network::GetGraph().