21.08
|
Private implementation of INetwork. More...
#include <Network.hpp>
Public Member Functions | |
NetworkImpl (NetworkOptions networkOptions={}) | |
~NetworkImpl () | |
const Graph & | GetGraph () const |
Status | PrintGraph () |
IConnectableLayer * | AddInputLayer (LayerBindingId id, const char *name=nullptr) |
IConnectableLayer * | AddAbsLayer (const char *name=nullptr) |
IConnectableLayer * | AddActivationLayer (const ActivationDescriptor &activationDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddAdditionLayer (const char *name=nullptr) |
IConnectableLayer * | AddArgMinMaxLayer (const ArgMinMaxDescriptor &desc, const char *name=nullptr) |
IConnectableLayer * | AddBatchNormalizationLayer (const BatchNormalizationDescriptor &desc, const ConstTensor &mean, const ConstTensor &variance, const ConstTensor &beta, const ConstTensor &gamma, const char *name=nullptr) |
IConnectableLayer * | AddBatchToSpaceNdLayer (const BatchToSpaceNdDescriptor &batchToSpaceNdDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddCastLayer (const char *name=nullptr) |
IConnectableLayer * | AddComparisonLayer (const ComparisonDescriptor &comparisonDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddConcatLayer (const ConcatDescriptor &concatDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddConvolution2dLayer (const Convolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr) |
IConnectableLayer * | AddConvolution2dLayer (const Convolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const char *name=nullptr) |
IConnectableLayer * | AddConvolution2dLayer (const Convolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const ConstTensor &biases, const char *name=nullptr) |
IConnectableLayer * | AddConstantLayer (const ConstTensor &input, const char *name=nullptr) |
IConnectableLayer * | AddDepthToSpaceLayer (const DepthToSpaceDescriptor &depthToSpaceDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddDepthwiseConvolution2dLayer (const DepthwiseConvolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr) |
IConnectableLayer * | AddDepthwiseConvolution2dLayer (const DepthwiseConvolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const char *name=nullptr) |
IConnectableLayer * | AddDepthwiseConvolution2dLayer (const DepthwiseConvolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const ConstTensor &biases, const char *name=nullptr) |
IConnectableLayer * | AddDequantizeLayer (const char *name=nullptr) |
IConnectableLayer * | AddDetectionPostProcessLayer (const DetectionPostProcessDescriptor &descriptor, const ConstTensor &anchors, const char *name=nullptr) |
IConnectableLayer * | AddDivisionLayer (const char *name=nullptr) |
IConnectableLayer * | AddElementwiseUnaryLayer (const ElementwiseUnaryDescriptor &elementwiseUnaryDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddEqualLayer (const char *name=nullptr) |
IConnectableLayer * | AddMergeLayer (const char *name=nullptr) |
IConnectableLayer * | AddFillLayer (const FillDescriptor &fillDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddFloorLayer (const char *name=nullptr) |
IConnectableLayer * | AddFullyConnectedLayer (const FullyConnectedDescriptor &fullyConnectedDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddFullyConnectedLayer (const FullyConnectedDescriptor &fullyConnectedDescriptor, const Optional< ConstTensor > &weights, const Optional< ConstTensor > &biases, const char *name=nullptr) |
IConnectableLayer * | AddFullyConnectedLayer (const FullyConnectedDescriptor &fullyConnectedDescriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr) |
IConnectableLayer * | AddGatherLayer (const char *name=nullptr) |
IConnectableLayer * | AddGatherLayer (const GatherDescriptor &gatherDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddGreaterLayer (const char *name=nullptr) |
IConnectableLayer * | AddInstanceNormalizationLayer (const InstanceNormalizationDescriptor &desc, const char *name=nullptr) |
IConnectableLayer * | AddL2NormalizationLayer (const L2NormalizationDescriptor &desc, const char *name=nullptr) |
IConnectableLayer * | AddLogSoftmaxLayer (const LogSoftmaxDescriptor &logSoftmaxDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddLogicalBinaryLayer (const LogicalBinaryDescriptor &logicalBinaryDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddLstmLayer (const LstmDescriptor &descriptor, const LstmInputParams ¶ms, const char *name=nullptr) |
IConnectableLayer * | AddMaximumLayer (const char *name=nullptr) |
IConnectableLayer * | AddMeanLayer (const MeanDescriptor &meanDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddMinimumLayer (const char *name=nullptr) |
IConnectableLayer * | AddMergerLayer (const MergerDescriptor &mergerDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddMultiplicationLayer (const char *name=nullptr) |
IConnectableLayer * | AddNormalizationLayer (const NormalizationDescriptor &normalizationDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddOutputLayer (LayerBindingId id, const char *name=nullptr) |
IConnectableLayer * | AddPadLayer (const PadDescriptor &padDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddPermuteLayer (const PermuteDescriptor &permuteDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddPooling2dLayer (const Pooling2dDescriptor &pooling2dDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddPreluLayer (const char *name=nullptr) |
IConnectableLayer * | AddQuantizeLayer (const char *name=nullptr) |
IConnectableLayer * | AddQLstmLayer (const QLstmDescriptor &descriptor, const LstmInputParams ¶ms, const char *name=nullptr) |
IConnectableLayer * | AddQuantizedLstmLayer (const QuantizedLstmInputParams ¶ms, const char *name=nullptr) |
IConnectableLayer * | AddRankLayer (const char *name=nullptr) |
IConnectableLayer * | AddReduceLayer (const ReduceDescriptor &reduceDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddResizeBilinearLayer (const ResizeBilinearDescriptor &resizeDesc, const char *name=nullptr) |
IConnectableLayer * | AddResizeLayer (const ResizeDescriptor &resizeDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddReshapeLayer (const ReshapeDescriptor &reshapeDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddRsqrtLayer (const char *name=nullptr) |
IConnectableLayer * | AddShapeLayer (const char *name=nullptr) |
IConnectableLayer * | AddSliceLayer (const SliceDescriptor &sliceDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddSoftmaxLayer (const SoftmaxDescriptor &softmaxDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddSplitterLayer (const ViewsDescriptor &splitterDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddSpaceToBatchNdLayer (const SpaceToBatchNdDescriptor &spaceToBatchNdDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddSpaceToDepthLayer (const SpaceToDepthDescriptor &spaceToDepthDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddStackLayer (const StackDescriptor &stackDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddStandInLayer (const StandInDescriptor &descriptor, const char *name=nullptr) |
IConnectableLayer * | AddStridedSliceLayer (const StridedSliceDescriptor &stridedSliceDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddSubtractionLayer (const char *name=nullptr) |
IConnectableLayer * | AddSwitchLayer (const char *name=nullptr) |
IConnectableLayer * | AddTransposeConvolution2dLayer (const TransposeConvolution2dDescriptor &descriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr) |
IConnectableLayer * | AddTransposeLayer (const TransposeDescriptor &transposeDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddUnidirectionalSequenceLstmLayer (const UnidirectionalSequenceLstmDescriptor &descriptor, const LstmInputParams ¶ms, const char *name=nullptr) |
void | Accept (ILayerVisitor &visitor) const |
void | ExecuteStrategy (IStrategy &strategy) const |
Private implementation of INetwork.
Definition at line 31 of file Network.hpp.
NetworkImpl | ( | NetworkOptions | networkOptions = {} | ) |
Definition at line 1790 of file Network.cpp.
~NetworkImpl | ( | ) |
Definition at line 1795 of file Network.cpp.
void Accept | ( | ILayerVisitor & | visitor | ) | const |
Definition at line 2824 of file Network.cpp.
References NetworkImpl::GetGraph().
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddAbsLayer | ( | const char * | name = nullptr | ) |
Definition at line 2114 of file Network.cpp.
References armnn::Abs, and NetworkImpl::AddElementwiseUnaryLayer().
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddActivationLayer | ( | const ActivationDescriptor & | activationDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2062 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().
IConnectableLayer * AddAdditionLayer | ( | const char * | name = nullptr | ) |
Definition at line 2119 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().
IConnectableLayer * AddArgMinMaxLayer | ( | const ArgMinMaxDescriptor & | desc, |
const char * | name = nullptr |
||
) |
Definition at line 2068 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddBatchNormalizationLayer | ( | const BatchNormalizationDescriptor & | desc, |
const ConstTensor & | mean, | ||
const ConstTensor & | variance, | ||
const ConstTensor & | beta, | ||
const ConstTensor & | gamma, | ||
const char * | name = nullptr |
||
) |
Definition at line 2134 of file Network.cpp.
References BatchNormalizationLayer::m_Mean.
Referenced by NetworkImpl::GetGraph(), TEST_SUITE(), and armnn::TEST_SUITE().
IConnectableLayer * AddBatchToSpaceNdLayer | ( | const BatchToSpaceNdDescriptor & | batchToSpaceNdDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 1810 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddCastLayer | ( | const char * | name = nullptr | ) |
Definition at line 1816 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddComparisonLayer | ( | const ComparisonDescriptor & | comparisonDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 1821 of file Network.cpp.
Referenced by NetworkImpl::AddEqualLayer(), NetworkImpl::AddGreaterLayer(), and NetworkImpl::GetGraph().
IConnectableLayer * AddConcatLayer | ( | const ConcatDescriptor & | concatDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 1930 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 NetworkImpl::AddMergerLayer(), NetworkImpl::GetGraph(), and TEST_SUITE().
IConnectableLayer * AddConstantLayer | ( | const ConstTensor & | input, |
const char * | name = nullptr |
||
) |
Definition at line 2204 of file Network.cpp.
References ConstantLayer::m_LayerOutput.
Referenced by NetworkImpl::GetGraph(), TEST_SUITE(), and armnn::TEST_SUITE().
IConnectableLayer * AddConvolution2dLayer | ( | const Convolution2dDescriptor & | convolution2dDescriptor, |
const ConstTensor & | weights, | ||
const Optional< ConstTensor > & | biases, | ||
const char * | name = nullptr |
||
) |
Definition at line 1958 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), TEST_SUITE(), and armnn::TEST_SUITE().
IConnectableLayer * AddConvolution2dLayer | ( | const Convolution2dDescriptor & | convolution2dDescriptor, |
const ConstTensor & | weights, | ||
const char * | name = nullptr |
||
) |
Definition at line 1966 of file Network.cpp.
IConnectableLayer * AddConvolution2dLayer | ( | const Convolution2dDescriptor & | convolution2dDescriptor, |
const ConstTensor & | weights, | ||
const ConstTensor & | biases, | ||
const char * | name = nullptr |
||
) |
Definition at line 1974 of file Network.cpp.
References OptionalBase::has_value(), DepthwiseConvolution2dDescriptor::m_BiasEnabled, DepthwiseConvolution2dLayer::m_Weight, and OptionalReferenceSwitch< std::is_reference< T >::value, T >::value().
IConnectableLayer * AddDepthToSpaceLayer | ( | const DepthToSpaceDescriptor & | depthToSpaceDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2006 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddDepthwiseConvolution2dLayer | ( | const DepthwiseConvolution2dDescriptor & | convolution2dDescriptor, |
const ConstTensor & | weights, | ||
const Optional< ConstTensor > & | biases, | ||
const char * | name = nullptr |
||
) |
Definition at line 2012 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and armnn::TEST_SUITE().
IConnectableLayer * AddDepthwiseConvolution2dLayer | ( | const DepthwiseConvolution2dDescriptor & | convolution2dDescriptor, |
const ConstTensor & | weights, | ||
const char * | name = nullptr |
||
) |
Definition at line 2021 of file Network.cpp.
IConnectableLayer * AddDepthwiseConvolution2dLayer | ( | const DepthwiseConvolution2dDescriptor & | convolution2dDescriptor, |
const ConstTensor & | weights, | ||
const ConstTensor & | biases, | ||
const char * | name = nullptr |
||
) |
Definition at line 2030 of file Network.cpp.
IConnectableLayer * AddDequantizeLayer | ( | const char * | name = nullptr | ) |
Definition at line 2402 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddDetectionPostProcessLayer | ( | const DetectionPostProcessDescriptor & | descriptor, |
const ConstTensor & | anchors, | ||
const char * | name = nullptr |
||
) |
Definition at line 2040 of file Network.cpp.
References DetectionPostProcessLayer::m_Anchors.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddDivisionLayer | ( | const char * | name = nullptr | ) |
Definition at line 2377 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddElementwiseUnaryLayer | ( | const ElementwiseUnaryDescriptor & | elementwiseUnaryDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 1827 of file Network.cpp.
Referenced by NetworkImpl::AddAbsLayer(), NetworkImpl::AddRsqrtLayer(), and NetworkImpl::GetGraph().
IConnectableLayer * AddEqualLayer | ( | const char * | name = nullptr | ) |
Definition at line 2418 of file Network.cpp.
References NetworkImpl::AddComparisonLayer(), and armnn::Equal.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddFillLayer | ( | const FillDescriptor & | fillDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 1833 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddFloorLayer | ( | const char * | name = nullptr | ) |
Definition at line 2231 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().
IConnectableLayer * AddFullyConnectedLayer | ( | const FullyConnectedDescriptor & | fullyConnectedDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 1839 of file Network.cpp.
Referenced by NetworkImpl::AddFullyConnectedLayer(), NetworkImpl::GetGraph(), TEST_SUITE(), and armnn::TEST_SUITE().
IConnectableLayer * AddFullyConnectedLayer | ( | const FullyConnectedDescriptor & | fullyConnectedDescriptor, |
const Optional< ConstTensor > & | weights, | ||
const Optional< ConstTensor > & | biases, | ||
const char * | name = nullptr |
||
) |
Definition at line 1845 of file Network.cpp.
References OutputSlot::Connect(), FullyConnectedDescriptor::GetNumInputs(), Layer::GetOutputSlot(), OptionalBase::has_value(), FullyConnectedDescriptor::m_BiasEnabled, FullyConnectedDescriptor::m_ConstantWeights, ConstantLayer::m_LayerOutput, OutputSlot::SetTensorInfo(), and OptionalReferenceSwitch< std::is_reference< T >::value, T >::value().
IConnectableLayer * AddFullyConnectedLayer | ( | const FullyConnectedDescriptor & | fullyConnectedDescriptor, |
const ConstTensor & | weights, | ||
const Optional< ConstTensor > & | biases, | ||
const char * | name = nullptr |
||
) |
Definition at line 1921 of file Network.cpp.
References NetworkImpl::AddFullyConnectedLayer().
IConnectableLayer * AddGatherLayer | ( | const char * | name = nullptr | ) |
Definition at line 2428 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddGatherLayer | ( | const GatherDescriptor & | gatherDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2434 of file Network.cpp.
IConnectableLayer * AddGreaterLayer | ( | const char * | name = nullptr | ) |
Definition at line 2413 of file Network.cpp.
References NetworkImpl::AddComparisonLayer(), and armnn::Greater.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddInputLayer | ( | LayerBindingId | id, |
const char * | name = nullptr |
||
) |
Definition at line 1805 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), armnn::TEST_SUITE(), and TEST_SUITE().
IConnectableLayer * AddInstanceNormalizationLayer | ( | const InstanceNormalizationDescriptor & | desc, |
const char * | name = nullptr |
||
) |
Definition at line 2186 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddL2NormalizationLayer | ( | const L2NormalizationDescriptor & | desc, |
const char * | name = nullptr |
||
) |
Definition at line 2192 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddLogicalBinaryLayer | ( | const LogicalBinaryDescriptor & | logicalBinaryDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2676 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddLogSoftmaxLayer | ( | const LogSoftmaxDescriptor & | logSoftmaxDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2198 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddLstmLayer | ( | const LstmDescriptor & | descriptor, |
const LstmInputParams & | params, | ||
const char * | name = nullptr |
||
) |
Definition at line 2236 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 NetworkImpl::GetGraph(), and armnn::TEST_SUITE().
IConnectableLayer * AddMaximumLayer | ( | const char * | name = nullptr | ) |
Definition at line 2098 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddMeanLayer | ( | const MeanDescriptor & | meanDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2387 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddMergeLayer | ( | const char * | name = nullptr | ) |
Definition at line 2440 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddMergerLayer | ( | const MergerDescriptor & | mergerDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2108 of file Network.cpp.
References NetworkImpl::AddConcatLayer().
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddMinimumLayer | ( | const char * | name = nullptr | ) |
Definition at line 2103 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddMultiplicationLayer | ( | const char * | name = nullptr | ) |
Definition at line 2124 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().
IConnectableLayer * AddNormalizationLayer | ( | const NormalizationDescriptor & | normalizationDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2074 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().
IConnectableLayer * AddOutputLayer | ( | LayerBindingId | id, |
const char * | name = nullptr |
||
) |
Definition at line 2129 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), armnn::TEST_SUITE(), and TEST_SUITE().
IConnectableLayer * AddPadLayer | ( | const PadDescriptor & | padDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2392 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddPermuteLayer | ( | const PermuteDescriptor & | permuteDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2050 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddPooling2dLayer | ( | const Pooling2dDescriptor & | pooling2dDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2056 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().
IConnectableLayer * AddPreluLayer | ( | const char * | name = nullptr | ) |
Definition at line 2450 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddQLstmLayer | ( | const QLstmDescriptor & | descriptor, |
const LstmInputParams & | params, | ||
const char * | name = nullptr |
||
) |
Definition at line 2534 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 NetworkImpl::GetGraph(), and armnn::TEST_SUITE().
IConnectableLayer * AddQuantizedLstmLayer | ( | const QuantizedLstmInputParams & | params, |
const char * | name = nullptr |
||
) |
Definition at line 2496 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 NetworkImpl::GetGraph(), and armnn::TEST_SUITE().
IConnectableLayer * AddQuantizeLayer | ( | const char * | name = nullptr | ) |
Definition at line 2397 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddRankLayer | ( | const char * | name = nullptr | ) |
Definition at line 2151 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddReduceLayer | ( | const ReduceDescriptor & | reduceDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2156 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddReshapeLayer | ( | const ReshapeDescriptor & | reshapeDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2213 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddResizeBilinearLayer | ( | const ResizeBilinearDescriptor & | resizeDesc, |
const char * | name = nullptr |
||
) |
Definition at line 2162 of file Network.cpp.
References armnn::Bilinear, ResizeBilinearDescriptor::m_AlignCorners, ResizeDescriptor::m_AlignCorners, ResizeBilinearDescriptor::m_DataLayout, ResizeDescriptor::m_DataLayout, ResizeBilinearDescriptor::m_HalfPixelCenters, ResizeDescriptor::m_HalfPixelCenters, ResizeDescriptor::m_Method, ResizeBilinearDescriptor::m_TargetHeight, ResizeDescriptor::m_TargetHeight, ResizeBilinearDescriptor::m_TargetWidth, and ResizeDescriptor::m_TargetWidth.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddResizeLayer | ( | const ResizeDescriptor & | resizeDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2176 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddRsqrtLayer | ( | const char * | name = nullptr | ) |
Definition at line 2423 of file Network.cpp.
References NetworkImpl::AddElementwiseUnaryLayer(), and armnn::Rsqrt.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddShapeLayer | ( | const char * | name = nullptr | ) |
Definition at line 2181 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddSliceLayer | ( | const SliceDescriptor & | sliceDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2081 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddSoftmaxLayer | ( | const SoftmaxDescriptor & | softmaxDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2086 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().
IConnectableLayer * AddSpaceToBatchNdLayer | ( | const SpaceToBatchNdDescriptor & | spaceToBatchNdDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2219 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddSpaceToDepthLayer | ( | const SpaceToDepthDescriptor & | spaceToDepthDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2225 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddSplitterLayer | ( | const ViewsDescriptor & | splitterDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2092 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().
IConnectableLayer * AddStackLayer | ( | const StackDescriptor & | stackDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2483 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddStandInLayer | ( | const StandInDescriptor & | descriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2490 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().
IConnectableLayer * AddStridedSliceLayer | ( | const StridedSliceDescriptor & | stridedSliceDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2407 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddSubtractionLayer | ( | const char * | name = nullptr | ) |
Definition at line 2382 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddSwitchLayer | ( | const char * | name = nullptr | ) |
Definition at line 2445 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddTransposeConvolution2dLayer | ( | const TransposeConvolution2dDescriptor & | descriptor, |
const ConstTensor & | weights, | ||
const Optional< ConstTensor > & | biases, | ||
const char * | name = nullptr |
||
) |
Definition at line 2455 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 NetworkImpl::GetGraph().
IConnectableLayer * AddTransposeLayer | ( | const TransposeDescriptor & | transposeDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2477 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddUnidirectionalSequenceLstmLayer | ( | const UnidirectionalSequenceLstmDescriptor & | descriptor, |
const LstmInputParams & | params, | ||
const char * | name = nullptr |
||
) |
Definition at line 2682 of file Network.cpp.
References UnidirectionalSequenceLstmLayer::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 NetworkImpl::GetGraph().
void ExecuteStrategy | ( | IStrategy & | strategy | ) | const |
Definition at line 2832 of file Network.cpp.
References NetworkImpl::GetGraph().
Referenced by NetworkImpl::GetGraph().
|
inline |
Definition at line 37 of file Network.hpp.
References NetworkImpl::Accept(), NetworkImpl::AddAbsLayer(), NetworkImpl::AddActivationLayer(), NetworkImpl::AddAdditionLayer(), NetworkImpl::AddArgMinMaxLayer(), NetworkImpl::AddBatchNormalizationLayer(), NetworkImpl::AddBatchToSpaceNdLayer(), NetworkImpl::AddCastLayer(), NetworkImpl::AddComparisonLayer(), NetworkImpl::AddConcatLayer(), NetworkImpl::AddConstantLayer(), NetworkImpl::AddConvolution2dLayer(), NetworkImpl::AddDepthToSpaceLayer(), NetworkImpl::AddDepthwiseConvolution2dLayer(), NetworkImpl::AddDequantizeLayer(), NetworkImpl::AddDetectionPostProcessLayer(), NetworkImpl::AddDivisionLayer(), NetworkImpl::AddElementwiseUnaryLayer(), NetworkImpl::AddEqualLayer(), NetworkImpl::AddFillLayer(), NetworkImpl::AddFloorLayer(), NetworkImpl::AddFullyConnectedLayer(), NetworkImpl::AddGatherLayer(), NetworkImpl::AddGreaterLayer(), NetworkImpl::AddInputLayer(), NetworkImpl::AddInstanceNormalizationLayer(), NetworkImpl::AddL2NormalizationLayer(), NetworkImpl::AddLogicalBinaryLayer(), NetworkImpl::AddLogSoftmaxLayer(), NetworkImpl::AddLstmLayer(), NetworkImpl::AddMaximumLayer(), NetworkImpl::AddMeanLayer(), NetworkImpl::AddMergeLayer(), NetworkImpl::AddMergerLayer(), NetworkImpl::AddMinimumLayer(), NetworkImpl::AddMultiplicationLayer(), NetworkImpl::AddNormalizationLayer(), NetworkImpl::AddOutputLayer(), NetworkImpl::AddPadLayer(), NetworkImpl::AddPermuteLayer(), NetworkImpl::AddPooling2dLayer(), NetworkImpl::AddPreluLayer(), NetworkImpl::AddQLstmLayer(), NetworkImpl::AddQuantizedLstmLayer(), NetworkImpl::AddQuantizeLayer(), NetworkImpl::AddRankLayer(), NetworkImpl::AddReduceLayer(), NetworkImpl::AddReshapeLayer(), NetworkImpl::AddResizeBilinearLayer(), NetworkImpl::AddResizeLayer(), NetworkImpl::AddRsqrtLayer(), NetworkImpl::AddShapeLayer(), NetworkImpl::AddSliceLayer(), NetworkImpl::AddSoftmaxLayer(), NetworkImpl::AddSpaceToBatchNdLayer(), NetworkImpl::AddSpaceToDepthLayer(), NetworkImpl::AddSplitterLayer(), NetworkImpl::AddStackLayer(), NetworkImpl::AddStandInLayer(), NetworkImpl::AddStridedSliceLayer(), NetworkImpl::AddSubtractionLayer(), NetworkImpl::AddSwitchLayer(), NetworkImpl::AddTransposeConvolution2dLayer(), NetworkImpl::AddTransposeLayer(), NetworkImpl::AddUnidirectionalSequenceLstmLayer(), ARMNN_DEPRECATED_MSG, NetworkImpl::ExecuteStrategy(), and NetworkImpl::PrintGraph().
Referenced by NetworkImpl::Accept(), NetworkImpl::ExecuteStrategy(), and TEST_SUITE().
Status PrintGraph | ( | ) |
Definition at line 1799 of file Network.cpp.
References armnn::Success.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().