21.05
|
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 * | AddArgMinMaxLayer (const ArgMinMaxDescriptor &desc, 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 * | 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 * | AddElementwiseUnaryLayer (const ElementwiseUnaryDescriptor &elementwiseUnaryDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddFillLayer (const FillDescriptor &fillDescriptor, 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 * | AddFullyConnectedLayer (const FullyConnectedDescriptor &fullyConnectedDescriptor, const ConstTensor &weights, const char *name=nullptr) |
IConnectableLayer * | AddFullyConnectedLayer (const FullyConnectedDescriptor &fullyConnectedDescriptor, const ConstTensor &weights, const ConstTensor &biases, const char *name=nullptr) |
IConnectableLayer * | AddGatherLayer (const char *name=nullptr) |
IConnectableLayer * | AddGatherLayer (const GatherDescriptor &gatherDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddPermuteLayer (const PermuteDescriptor &permuteDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddPooling2dLayer (const Pooling2dDescriptor &pooling2dDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddActivationLayer (const ActivationDescriptor &activationDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddNormalizationLayer (const NormalizationDescriptor &normalizationDescriptor, 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 * | AddMergerLayer (const MergerDescriptor &mergerDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddAbsLayer (const char *name=nullptr) |
IConnectableLayer * | AddAdditionLayer (const char *name=nullptr) |
IConnectableLayer * | AddMultiplicationLayer (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 * | AddRankLayer (const char *name=nullptr) |
IConnectableLayer * | AddResizeBilinearLayer (const ResizeBilinearDescriptor &resizeDesc, const char *name=nullptr) |
IConnectableLayer * | AddResizeLayer (const ResizeDescriptor &resizeDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddReduceLayer (const ReduceDescriptor &reduceDescriptor, 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 * | AddConstantLayer (const ConstTensor &input, const char *name=nullptr) |
IConnectableLayer * | AddReshapeLayer (const ReshapeDescriptor &reshapeDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddSpaceToBatchNdLayer (const SpaceToBatchNdDescriptor &spaceToBatchNdDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddSpaceToDepthLayer (const SpaceToDepthDescriptor &spaceToDepthDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddFloorLayer (const char *name=nullptr) |
IConnectableLayer * | AddOutputLayer (LayerBindingId id, const char *name=nullptr) |
IConnectableLayer * | AddLstmLayer (const LstmDescriptor &descriptor, const LstmInputParams ¶ms, const char *name=nullptr) |
IConnectableLayer * | AddDivisionLayer (const char *name=nullptr) |
IConnectableLayer * | AddSubtractionLayer (const char *name=nullptr) |
IConnectableLayer * | AddMaximumLayer (const char *name=nullptr) |
IConnectableLayer * | AddMeanLayer (const MeanDescriptor &meanDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddPadLayer (const PadDescriptor &padDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddQuantizeLayer (const char *name=nullptr) |
IConnectableLayer * | AddStridedSliceLayer (const StridedSliceDescriptor &stridedSliceDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddMinimumLayer (const char *name=nullptr) |
IConnectableLayer * | AddGreaterLayer (const char *name=nullptr) |
IConnectableLayer * | AddEqualLayer (const char *name=nullptr) |
IConnectableLayer * | AddRsqrtLayer (const char *name=nullptr) |
IConnectableLayer * | AddMergeLayer (const char *name=nullptr) |
IConnectableLayer * | AddSwitchLayer (const char *name=nullptr) |
IConnectableLayer * | AddPreluLayer (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 * | AddStackLayer (const StackDescriptor &stackDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddStandInLayer (const StandInDescriptor &descriptor, 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 * | AddLogicalBinaryLayer (const LogicalBinaryDescriptor &logicalBinaryDescriptor, 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 1734 of file Network.cpp.
~NetworkImpl | ( | ) |
Definition at line 1739 of file Network.cpp.
void Accept | ( | ILayerVisitor & | visitor | ) | const |
Definition at line 2595 of file Network.cpp.
References NetworkImpl::GetGraph().
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddAbsLayer | ( | const char * | name = nullptr | ) |
Definition at line 2032 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 1980 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().
IConnectableLayer * AddAdditionLayer | ( | const char * | name = nullptr | ) |
Definition at line 2037 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().
IConnectableLayer * AddArgMinMaxLayer | ( | const ArgMinMaxDescriptor & | desc, |
const char * | name = nullptr |
||
) |
Definition at line 1986 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 2052 of file Network.cpp.
References BatchNormalizationLayer::m_Mean.
Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().
IConnectableLayer * AddBatchToSpaceNdLayer | ( | const BatchToSpaceNdDescriptor & | batchToSpaceNdDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 1754 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddCastLayer | ( | const char * | name = nullptr | ) |
Definition at line 1760 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddComparisonLayer | ( | const ComparisonDescriptor & | comparisonDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 1765 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 1848 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(), BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().
IConnectableLayer * AddConstantLayer | ( | const ConstTensor & | input, |
const char * | name = nullptr |
||
) |
Definition at line 2117 of file Network.cpp.
References ConstantLayer::m_LayerOutput.
Referenced by armnn::BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().
IConnectableLayer * AddConvolution2dLayer | ( | const Convolution2dDescriptor & | convolution2dDescriptor, |
const ConstTensor & | weights, | ||
const Optional< ConstTensor > & | biases, | ||
const char * | name = nullptr |
||
) |
Definition at line 1876 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().
IConnectableLayer * AddConvolution2dLayer | ( | const Convolution2dDescriptor & | convolution2dDescriptor, |
const ConstTensor & | weights, | ||
const char * | name = nullptr |
||
) |
Definition at line 1884 of file Network.cpp.
IConnectableLayer * AddConvolution2dLayer | ( | const Convolution2dDescriptor & | convolution2dDescriptor, |
const ConstTensor & | weights, | ||
const ConstTensor & | biases, | ||
const char * | name = nullptr |
||
) |
Definition at line 1892 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 1924 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 1930 of file Network.cpp.
Referenced by armnn::BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().
IConnectableLayer * AddDepthwiseConvolution2dLayer | ( | const DepthwiseConvolution2dDescriptor & | convolution2dDescriptor, |
const ConstTensor & | weights, | ||
const char * | name = nullptr |
||
) |
Definition at line 1939 of file Network.cpp.
IConnectableLayer * AddDepthwiseConvolution2dLayer | ( | const DepthwiseConvolution2dDescriptor & | convolution2dDescriptor, |
const ConstTensor & | weights, | ||
const ConstTensor & | biases, | ||
const char * | name = nullptr |
||
) |
Definition at line 1948 of file Network.cpp.
IConnectableLayer * AddDequantizeLayer | ( | const char * | name = nullptr | ) |
Definition at line 2315 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddDetectionPostProcessLayer | ( | const DetectionPostProcessDescriptor & | descriptor, |
const ConstTensor & | anchors, | ||
const char * | name = nullptr |
||
) |
Definition at line 1958 of file Network.cpp.
References anchors(), and DetectionPostProcessLayer::m_Anchors.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddDivisionLayer | ( | const char * | name = nullptr | ) |
Definition at line 2290 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddElementwiseUnaryLayer | ( | const ElementwiseUnaryDescriptor & | elementwiseUnaryDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 1771 of file Network.cpp.
Referenced by NetworkImpl::AddAbsLayer(), NetworkImpl::AddRsqrtLayer(), and NetworkImpl::GetGraph().
IConnectableLayer * AddEqualLayer | ( | const char * | name = nullptr | ) |
Definition at line 2331 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 1777 of file Network.cpp.
References OptionalBase::has_value(), FullyConnectedDescriptor::m_BiasEnabled, FullyConnectedDescriptor::m_ConstantWeights, FullyConnectedLayer::m_Weight, and OptionalReferenceSwitch< std::is_reference< T >::value, T >::value().
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddFloorLayer | ( | const char * | name = nullptr | ) |
Definition at line 2144 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().
IConnectableLayer * AddFullyConnectedLayer | ( | const FullyConnectedDescriptor & | fullyConnectedDescriptor, |
const Optional< ConstTensor > & | weights, | ||
const Optional< ConstTensor > & | biases, | ||
const char * | name = nullptr |
||
) |
Definition at line 1812 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().
IConnectableLayer * AddFullyConnectedLayer | ( | const FullyConnectedDescriptor & | fullyConnectedDescriptor, |
const ConstTensor & | weights, | ||
const Optional< ConstTensor > & | biases, | ||
const char * | name = nullptr |
||
) |
Definition at line 1820 of file Network.cpp.
IConnectableLayer * AddFullyConnectedLayer | ( | const FullyConnectedDescriptor & | fullyConnectedDescriptor, |
const ConstTensor & | weights, | ||
const char * | name = nullptr |
||
) |
Definition at line 1829 of file Network.cpp.
IConnectableLayer * AddFullyConnectedLayer | ( | const FullyConnectedDescriptor & | fullyConnectedDescriptor, |
const ConstTensor & | weights, | ||
const ConstTensor & | biases, | ||
const char * | name = nullptr |
||
) |
Definition at line 1838 of file Network.cpp.
IConnectableLayer * AddGatherLayer | ( | const char * | name = nullptr | ) |
Definition at line 2341 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddGatherLayer | ( | const GatherDescriptor & | gatherDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2347 of file Network.cpp.
IConnectableLayer * AddGreaterLayer | ( | const char * | name = nullptr | ) |
Definition at line 2326 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 1749 of file Network.cpp.
Referenced by armnn::BOOST_AUTO_TEST_CASE(), BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().
IConnectableLayer * AddInstanceNormalizationLayer | ( | const InstanceNormalizationDescriptor & | desc, |
const char * | name = nullptr |
||
) |
Definition at line 2099 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddL2NormalizationLayer | ( | const L2NormalizationDescriptor & | desc, |
const char * | name = nullptr |
||
) |
Definition at line 2105 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddLogicalBinaryLayer | ( | const LogicalBinaryDescriptor & | logicalBinaryDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2589 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddLogSoftmaxLayer | ( | const LogSoftmaxDescriptor & | logSoftmaxDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2111 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddLstmLayer | ( | const LstmDescriptor & | descriptor, |
const LstmInputParams & | params, | ||
const char * | name = nullptr |
||
) |
Definition at line 2149 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 NetworkImpl::GetGraph().
IConnectableLayer * AddMaximumLayer | ( | const char * | name = nullptr | ) |
Definition at line 2016 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddMeanLayer | ( | const MeanDescriptor & | meanDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2300 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddMergeLayer | ( | const char * | name = nullptr | ) |
Definition at line 2353 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddMergerLayer | ( | const MergerDescriptor & | mergerDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2026 of file Network.cpp.
References NetworkImpl::AddConcatLayer().
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddMinimumLayer | ( | const char * | name = nullptr | ) |
Definition at line 2021 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddMultiplicationLayer | ( | const char * | name = nullptr | ) |
Definition at line 2042 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().
IConnectableLayer * AddNormalizationLayer | ( | const NormalizationDescriptor & | normalizationDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 1992 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().
IConnectableLayer * AddOutputLayer | ( | LayerBindingId | id, |
const char * | name = nullptr |
||
) |
Definition at line 2047 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().
IConnectableLayer * AddPadLayer | ( | const PadDescriptor & | padDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2305 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddPermuteLayer | ( | const PermuteDescriptor & | permuteDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 1968 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddPooling2dLayer | ( | const Pooling2dDescriptor & | pooling2dDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 1974 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().
IConnectableLayer * AddPreluLayer | ( | const char * | name = nullptr | ) |
Definition at line 2363 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddQLstmLayer | ( | const QLstmDescriptor & | descriptor, |
const LstmInputParams & | params, | ||
const char * | name = nullptr |
||
) |
Definition at line 2447 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 NetworkImpl::GetGraph().
IConnectableLayer * AddQuantizedLstmLayer | ( | const QuantizedLstmInputParams & | params, |
const char * | name = nullptr |
||
) |
Definition at line 2409 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 NetworkImpl::GetGraph().
IConnectableLayer * AddQuantizeLayer | ( | const char * | name = nullptr | ) |
Definition at line 2310 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddRankLayer | ( | const char * | name = nullptr | ) |
Definition at line 2069 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddReduceLayer | ( | const ReduceDescriptor & | reduceDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2074 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddReshapeLayer | ( | const ReshapeDescriptor & | reshapeDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2126 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddResizeBilinearLayer | ( | const ResizeBilinearDescriptor & | resizeDesc, |
const char * | name = nullptr |
||
) |
Definition at line 2080 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 2094 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddRsqrtLayer | ( | const char * | name = nullptr | ) |
Definition at line 2336 of file Network.cpp.
References NetworkImpl::AddElementwiseUnaryLayer(), and armnn::Rsqrt.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddSliceLayer | ( | const SliceDescriptor & | sliceDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 1999 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddSoftmaxLayer | ( | const SoftmaxDescriptor & | softmaxDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2004 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().
IConnectableLayer * AddSpaceToBatchNdLayer | ( | const SpaceToBatchNdDescriptor & | spaceToBatchNdDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2132 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddSpaceToDepthLayer | ( | const SpaceToDepthDescriptor & | spaceToDepthDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2138 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddSplitterLayer | ( | const ViewsDescriptor & | splitterDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2010 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().
IConnectableLayer * AddStackLayer | ( | const StackDescriptor & | stackDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2396 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddStandInLayer | ( | const StandInDescriptor & | descriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2403 of file Network.cpp.
Referenced by BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().
IConnectableLayer * AddStridedSliceLayer | ( | const StridedSliceDescriptor & | stridedSliceDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2320 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddSubtractionLayer | ( | const char * | name = nullptr | ) |
Definition at line 2295 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddSwitchLayer | ( | const char * | name = nullptr | ) |
Definition at line 2358 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 2368 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 2390 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
void ExecuteStrategy | ( | IStrategy & | strategy | ) | const |
Definition at line 2603 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::AddSliceLayer(), NetworkImpl::AddSoftmaxLayer(), NetworkImpl::AddSpaceToBatchNdLayer(), NetworkImpl::AddSpaceToDepthLayer(), NetworkImpl::AddSplitterLayer(), NetworkImpl::AddStackLayer(), NetworkImpl::AddStandInLayer(), NetworkImpl::AddStridedSliceLayer(), NetworkImpl::AddSubtractionLayer(), NetworkImpl::AddSwitchLayer(), NetworkImpl::AddTransposeConvolution2dLayer(), NetworkImpl::AddTransposeLayer(), anchors(), ARMNN_DEPRECATED_MSG, NetworkImpl::ExecuteStrategy(), and NetworkImpl::PrintGraph().
Referenced by NetworkImpl::Accept(), BOOST_AUTO_TEST_CASE(), and NetworkImpl::ExecuteStrategy().
Status PrintGraph | ( | ) |
Definition at line 1743 of file Network.cpp.
References armnn::Success.
Referenced by BOOST_AUTO_TEST_CASE(), and NetworkImpl::GetGraph().