22.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 * | AddActivationLayer (const ActivationDescriptor &activationDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddAdditionLayer (const char *name=nullptr) |
IConnectableLayer * | AddArgMinMaxLayer (const ArgMinMaxDescriptor &desc, const char *name=nullptr) |
IConnectableLayer * | AddBatchMatMulLayer (const BatchMatMulDescriptor &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 * | AddChannelShuffleLayer (const ChannelShuffleDescriptor &channelShuffleDescriptor, 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 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 * | AddConvolution3dLayer (const Convolution3dDescriptor &convolution3dDescriptor, 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 char *name=nullptr) |
IConnectableLayer * | AddDepthwiseConvolution2dLayer (const DepthwiseConvolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const Optional< 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 * | 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 * | AddGatherLayer (const GatherDescriptor &gatherDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddGatherNdLayer (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 * | 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 * | AddPooling3dLayer (const Pooling3dDescriptor &pooling3dDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddPrecompiledLayer (const PreCompiledDescriptor &preCompiledDescriptor, CompiledBlobPtr compiledBlobPtr, const Optional< BackendId > &backend, 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 * | AddResizeLayer (const ResizeDescriptor &resizeDescriptor, const char *name=nullptr) |
IConnectableLayer * | AddReshapeLayer (const ReshapeDescriptor &reshapeDescriptor, 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) |
IConnectableLayer * | AddConvertFp16ToFp32Layer (const char *name=nullptr) |
IConnectableLayer * | AddConvertFp32ToFp16Layer (const char *name=nullptr) |
void | ExecuteStrategy (IStrategy &strategy) const |
Private implementation of INetwork.
Definition at line 31 of file Network.hpp.
NetworkImpl | ( | NetworkOptions | networkOptions = {} | ) |
Definition at line 1905 of file Network.cpp.
~NetworkImpl | ( | ) |
Definition at line 1910 of file Network.cpp.
IConnectableLayer * AddActivationLayer | ( | const ActivationDescriptor & | activationDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2168 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().
IConnectableLayer * AddAdditionLayer | ( | const char * | name = nullptr | ) |
Definition at line 2214 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().
IConnectableLayer * AddArgMinMaxLayer | ( | const ArgMinMaxDescriptor & | desc, |
const char * | name = nullptr |
||
) |
Definition at line 2174 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddBatchMatMulLayer | ( | const BatchMatMulDescriptor & | desc, |
const char * | name = nullptr |
||
) |
Definition at line 2889 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 2229 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 1925 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddCastLayer | ( | const char * | name = nullptr | ) |
Definition at line 1931 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddChannelShuffleLayer | ( | const ChannelShuffleDescriptor & | channelShuffleDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 1935 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddComparisonLayer | ( | const ComparisonDescriptor & | comparisonDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 1941 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddConcatLayer | ( | const ConcatDescriptor & | concatDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2041 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().
IConnectableLayer * AddConstantLayer | ( | const ConstTensor & | input, |
const char * | name = nullptr |
||
) |
Definition at line 2285 of file Network.cpp.
References ConstantLayer::m_LayerOutput.
Referenced by NetworkImpl::GetGraph(), TEST_SUITE(), and armnn::TEST_SUITE().
IConnectableLayer * AddConvertFp16ToFp32Layer | ( | const char * | name = nullptr | ) |
Definition at line 2079 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().
IConnectableLayer * AddConvertFp32ToFp16Layer | ( | const char * | name = nullptr | ) |
Definition at line 2084 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().
IConnectableLayer * AddConvolution2dLayer | ( | const Convolution2dDescriptor & | convolution2dDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2047 of file Network.cpp.
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 2053 of file Network.cpp.
References OutputSlot::Connect(), Layer::GetOutputSlot(), OptionalBase::has_value(), ConstantLayer::m_LayerOutput, OutputSlot::SetTensorInfo(), and OptionalReferenceSwitch< std::is_reference< T >::value, T >::value().
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 * AddConvolution3dLayer | ( | const Convolution3dDescriptor & | convolution3dDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2089 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddDepthToSpaceLayer | ( | const DepthToSpaceDescriptor & | depthToSpaceDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2095 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddDepthwiseConvolution2dLayer | ( | const DepthwiseConvolution2dDescriptor & | convolution2dDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2101 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and armnn::TEST_SUITE().
IConnectableLayer * AddDepthwiseConvolution2dLayer | ( | const DepthwiseConvolution2dDescriptor & | convolution2dDescriptor, |
const ConstTensor & | weights, | ||
const Optional< ConstTensor > & | biases, | ||
const char * | name = nullptr |
||
) |
Definition at line 2108 of file Network.cpp.
References OutputSlot::Connect(), Layer::GetOutputSlot(), OptionalBase::has_value(), ConstantLayer::m_LayerOutput, OutputSlot::SetTensorInfo(), and OptionalReferenceSwitch< std::is_reference< T >::value, T >::value().
IConnectableLayer * AddDequantizeLayer | ( | const char * | name = nullptr | ) |
Definition at line 2483 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddDetectionPostProcessLayer | ( | const DetectionPostProcessDescriptor & | descriptor, |
const ConstTensor & | anchors, | ||
const char * | name = nullptr |
||
) |
Definition at line 2140 of file Network.cpp.
References DetectionPostProcessLayer::m_Anchors.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddDivisionLayer | ( | const char * | name = nullptr | ) |
Definition at line 2458 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddElementwiseUnaryLayer | ( | const ElementwiseUnaryDescriptor & | elementwiseUnaryDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 1947 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddFillLayer | ( | const FillDescriptor & | fillDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 1953 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddFloorLayer | ( | const char * | name = nullptr | ) |
Definition at line 2312 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().
IConnectableLayer * AddFullyConnectedLayer | ( | const FullyConnectedDescriptor & | fullyConnectedDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 1959 of file Network.cpp.
Referenced by 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 1965 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 * AddGatherLayer | ( | const GatherDescriptor & | gatherDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2494 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddGatherNdLayer | ( | const char * | name = nullptr | ) |
Definition at line 2500 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddInputLayer | ( | LayerBindingId | id, |
const char * | name = nullptr |
||
) |
Definition at line 1920 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 2267 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddL2NormalizationLayer | ( | const L2NormalizationDescriptor & | desc, |
const char * | name = nullptr |
||
) |
Definition at line 2273 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddLogicalBinaryLayer | ( | const LogicalBinaryDescriptor & | logicalBinaryDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2741 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddLogSoftmaxLayer | ( | const LogSoftmaxDescriptor & | logSoftmaxDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2279 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddLstmLayer | ( | const LstmDescriptor & | descriptor, |
const LstmInputParams & | params, | ||
const char * | name = nullptr |
||
) |
Definition at line 2317 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 2204 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddMeanLayer | ( | const MeanDescriptor & | meanDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2468 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddMergeLayer | ( | const char * | name = nullptr | ) |
Definition at line 2505 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddMinimumLayer | ( | const char * | name = nullptr | ) |
Definition at line 2209 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddMultiplicationLayer | ( | const char * | name = nullptr | ) |
Definition at line 2219 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().
IConnectableLayer * AddNormalizationLayer | ( | const NormalizationDescriptor & | normalizationDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2180 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().
IConnectableLayer * AddOutputLayer | ( | LayerBindingId | id, |
const char * | name = nullptr |
||
) |
Definition at line 2224 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 2473 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddPermuteLayer | ( | const PermuteDescriptor & | permuteDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2150 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddPooling2dLayer | ( | const Pooling2dDescriptor & | pooling2dDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2156 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().
IConnectableLayer * AddPooling3dLayer | ( | const Pooling3dDescriptor & | pooling3dDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2162 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddPrecompiledLayer | ( | const PreCompiledDescriptor & | preCompiledDescriptor, |
CompiledBlobPtr | compiledBlobPtr, | ||
const Optional< BackendId > & | backend, | ||
const char * | name = nullptr |
||
) |
Definition at line 2894 of file Network.cpp.
References Layer::GetBackendHint(), OptionalBase::has_value(), Layer::SetBackendId(), PreCompiledLayer::SetPreCompiledObject(), and OptionalReferenceSwitch< std::is_reference< T >::value, T >::value().
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddPreluLayer | ( | const char * | name = nullptr | ) |
Definition at line 2515 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddQLstmLayer | ( | const QLstmDescriptor & | descriptor, |
const LstmInputParams & | params, | ||
const char * | name = nullptr |
||
) |
Definition at line 2599 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 2561 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 2478 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddRankLayer | ( | const char * | name = nullptr | ) |
Definition at line 2246 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddReduceLayer | ( | const ReduceDescriptor & | reduceDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2251 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddReshapeLayer | ( | const ReshapeDescriptor & | reshapeDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2294 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddResizeLayer | ( | const ResizeDescriptor & | resizeDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2257 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddShapeLayer | ( | const char * | name = nullptr | ) |
Definition at line 2262 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddSliceLayer | ( | const SliceDescriptor & | sliceDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2187 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddSoftmaxLayer | ( | const SoftmaxDescriptor & | softmaxDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2192 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().
IConnectableLayer * AddSpaceToBatchNdLayer | ( | const SpaceToBatchNdDescriptor & | spaceToBatchNdDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2300 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddSpaceToDepthLayer | ( | const SpaceToDepthDescriptor & | spaceToDepthDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2306 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddSplitterLayer | ( | const ViewsDescriptor & | splitterDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2198 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().
IConnectableLayer * AddStackLayer | ( | const StackDescriptor & | stackDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2548 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddStandInLayer | ( | const StandInDescriptor & | descriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2555 of file Network.cpp.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().
IConnectableLayer * AddStridedSliceLayer | ( | const StridedSliceDescriptor & | stridedSliceDescriptor, |
const char * | name = nullptr |
||
) |
Definition at line 2488 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddSubtractionLayer | ( | const char * | name = nullptr | ) |
Definition at line 2463 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddSwitchLayer | ( | const char * | name = nullptr | ) |
Definition at line 2510 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 2520 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 2542 of file Network.cpp.
Referenced by NetworkImpl::GetGraph().
IConnectableLayer * AddUnidirectionalSequenceLstmLayer | ( | const UnidirectionalSequenceLstmDescriptor & | descriptor, |
const LstmInputParams & | params, | ||
const char * | name = nullptr |
||
) |
Definition at line 2747 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 2927 of file Network.cpp.
References NetworkImpl::GetGraph().
Referenced by NetworkImpl::GetGraph().
|
inline |
Definition at line 37 of file Network.hpp.
References NetworkImpl::AddActivationLayer(), NetworkImpl::AddAdditionLayer(), NetworkImpl::AddArgMinMaxLayer(), NetworkImpl::AddBatchMatMulLayer(), NetworkImpl::AddBatchNormalizationLayer(), NetworkImpl::AddBatchToSpaceNdLayer(), NetworkImpl::AddCastLayer(), NetworkImpl::AddChannelShuffleLayer(), NetworkImpl::AddComparisonLayer(), NetworkImpl::AddConcatLayer(), NetworkImpl::AddConstantLayer(), NetworkImpl::AddConvertFp16ToFp32Layer(), NetworkImpl::AddConvertFp32ToFp16Layer(), NetworkImpl::AddConvolution2dLayer(), NetworkImpl::AddConvolution3dLayer(), NetworkImpl::AddDepthToSpaceLayer(), NetworkImpl::AddDepthwiseConvolution2dLayer(), NetworkImpl::AddDequantizeLayer(), NetworkImpl::AddDetectionPostProcessLayer(), NetworkImpl::AddDivisionLayer(), NetworkImpl::AddElementwiseUnaryLayer(), NetworkImpl::AddFillLayer(), NetworkImpl::AddFloorLayer(), NetworkImpl::AddFullyConnectedLayer(), NetworkImpl::AddGatherLayer(), NetworkImpl::AddGatherNdLayer(), NetworkImpl::AddInputLayer(), NetworkImpl::AddInstanceNormalizationLayer(), NetworkImpl::AddL2NormalizationLayer(), NetworkImpl::AddLogicalBinaryLayer(), NetworkImpl::AddLogSoftmaxLayer(), NetworkImpl::AddLstmLayer(), NetworkImpl::AddMaximumLayer(), NetworkImpl::AddMeanLayer(), NetworkImpl::AddMergeLayer(), NetworkImpl::AddMinimumLayer(), NetworkImpl::AddMultiplicationLayer(), NetworkImpl::AddNormalizationLayer(), NetworkImpl::AddOutputLayer(), NetworkImpl::AddPadLayer(), NetworkImpl::AddPermuteLayer(), NetworkImpl::AddPooling2dLayer(), NetworkImpl::AddPooling3dLayer(), NetworkImpl::AddPrecompiledLayer(), NetworkImpl::AddPreluLayer(), NetworkImpl::AddQLstmLayer(), NetworkImpl::AddQuantizedLstmLayer(), NetworkImpl::AddQuantizeLayer(), NetworkImpl::AddRankLayer(), NetworkImpl::AddReduceLayer(), NetworkImpl::AddReshapeLayer(), NetworkImpl::AddResizeLayer(), 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, armnn::ARMNN_DEPRECATED_MSG_REMOVAL_DATE(), NetworkImpl::ExecuteStrategy(), and NetworkImpl::PrintGraph().
Referenced by NetworkImpl::ExecuteStrategy(), and TEST_SUITE().
Status PrintGraph | ( | ) |
Definition at line 1914 of file Network.cpp.
References armnn::Success.
Referenced by NetworkImpl::GetGraph(), and TEST_SUITE().