21.11
|
#include <Layer.hpp>
Public Member Functions | |
InputSlot (Layer &owner, unsigned int slotIndex) | |
~InputSlot () | |
Layer & | GetOwningLayer () const |
unsigned int | GetSlotIndex () const |
const OutputSlot * | GetConnectedOutputSlot () const |
OutputSlot * | GetConnectedOutputSlot () |
void | SetConnection (OutputSlot *source) |
Links the slot to an output slot or breaks an existing link if passing nullptr. More... | |
void | Insert (Layer &layer) |
const IOutputSlot * | GetConnection () const override |
IOutputSlot * | GetConnection () override |
Additional Inherited Members | |
Protected Member Functions inherited from IInputSlot | |
~IInputSlot () | |
Not user deletable. More... | |
Definition at line 44 of file Layer.hpp.
References InputSlot::~InputSlot().
|
inline |
|
inline |
Definition at line 55 of file Layer.hpp.
Referenced by ConcatLayer::CreateWorkload(), DebugLayer::CreateWorkload(), armnn::optimizations::pad_fold::FoldPadIntoLayer2dImpl(), InputSlot::GetConnection(), LoadedNetwork::ImportOutputs(), Graph::InferTensorInfos(), InputSlot::Insert(), armnn::InsertConvertBf16ToFp32LayersBefore(), armnn::InsertConvertFp16ToFp32LayersBefore(), armnn::InsertConvertFp32ToBf16LayersAfter(), armnn::InsertConvertFp32ToBf16LayersBefore(), armnn::InsertConvertFp32ToFp16LayersAfter(), armnn::InsertDebugLayerAfter(), Graph::InsertNewLayer(), NeonBackend::OptimizeSubgraphView(), ClBackend::OptimizeSubgraphView(), LoadedNetwork::RegisterDebugCallback(), OptimizeConsecutiveReshapesImpl::Run(), PermuteAndBatchToSpaceAsDepthToSpaceImpl< PermuteType >::Run(), OptimizeInverseConversionsImpl::Run(), MovePermuteUpImpl::Run(), MoveTransposeUpImpl::Run(), OptimizeInversePermutesImpl< PermuteType >::Run(), SquashEqualSiblingsImpl< Comparable >::Run(), AddBroadcastReshapeLayerImpl::Run(), FuseBatchNorm< ConvLayer, ArmnnType, T >::Run(), and TEST_SUITE().
|
inline |
|
inlineoverridevirtual |
Implements IInputSlot.
Definition at line 199 of file Layer.hpp.
References InputSlot::GetConnectedOutputSlot().
Referenced by OutputSlot::DisconnectAll(), Layer::GetDataType(), OutputSlot::GetNumConnections(), OutputSlot::MoveAllConnections(), OutputSlot::operator==(), PermuteAndBatchToSpaceAsDepthToSpaceImpl< PermuteType >::Run(), Convolution3dLayer::SerializeLayerParameters(), DepthwiseConvolution2dLayer::SerializeLayerParameters(), Convolution2dLayer::SerializeLayerParameters(), InputSlot::SetConnection(), Graph::SubstituteSubgraph(), TEST_SUITE(), ElementwiseBaseLayer::ValidateTensorShapesFromInputs(), QuantizeLayer::ValidateTensorShapesFromInputs(), ActivationLayer::ValidateTensorShapesFromInputs(), FillLayer::ValidateTensorShapesFromInputs(), ConvertFp32ToFp16Layer::ValidateTensorShapesFromInputs(), ReduceLayer::ValidateTensorShapesFromInputs(), AbsLayer::ValidateTensorShapesFromInputs(), ConvertBf16ToFp32Layer::ValidateTensorShapesFromInputs(), FloorLayer::ValidateTensorShapesFromInputs(), ConvertFp16ToFp32Layer::ValidateTensorShapesFromInputs(), ConvertFp32ToBf16Layer::ValidateTensorShapesFromInputs(), InstanceNormalizationLayer::ValidateTensorShapesFromInputs(), L2NormalizationLayer::ValidateTensorShapesFromInputs(), SwitchLayer::ValidateTensorShapesFromInputs(), MemCopyLayer::ValidateTensorShapesFromInputs(), MemImportLayer::ValidateTensorShapesFromInputs(), MergeLayer::ValidateTensorShapesFromInputs(), NormalizationLayer::ValidateTensorShapesFromInputs(), ShapeLayer::ValidateTensorShapesFromInputs(), BatchToSpaceNdLayer::ValidateTensorShapesFromInputs(), DebugLayer::ValidateTensorShapesFromInputs(), Pooling2dLayer::ValidateTensorShapesFromInputs(), SliceLayer::ValidateTensorShapesFromInputs(), CastLayer::ValidateTensorShapesFromInputs(), ResizeLayer::ValidateTensorShapesFromInputs(), RsqrtLayer::ValidateTensorShapesFromInputs(), DequantizeLayer::ValidateTensorShapesFromInputs(), FakeQuantizationLayer::ValidateTensorShapesFromInputs(), SoftmaxLayer::ValidateTensorShapesFromInputs(), StackLayer::ValidateTensorShapesFromInputs(), TransposeLayer::ValidateTensorShapesFromInputs(), LogSoftmaxLayer::ValidateTensorShapesFromInputs(), MeanLayer::ValidateTensorShapesFromInputs(), PadLayer::ValidateTensorShapesFromInputs(), ChannelShuffleLayer::ValidateTensorShapesFromInputs(), PermuteLayer::ValidateTensorShapesFromInputs(), ReshapeLayer::ValidateTensorShapesFromInputs(), Convolution3dLayer::ValidateTensorShapesFromInputs(), StridedSliceLayer::ValidateTensorShapesFromInputs(), ElementwiseUnaryLayer::ValidateTensorShapesFromInputs(), ArgMinMaxLayer::ValidateTensorShapesFromInputs(), GatherLayer::ValidateTensorShapesFromInputs(), TransposeConvolution2dLayer::ValidateTensorShapesFromInputs(), PreluLayer::ValidateTensorShapesFromInputs(), ComparisonLayer::ValidateTensorShapesFromInputs(), DepthwiseConvolution2dLayer::ValidateTensorShapesFromInputs(), LogicalBinaryLayer::ValidateTensorShapesFromInputs(), DepthToSpaceLayer::ValidateTensorShapesFromInputs(), SpaceToBatchNdLayer::ValidateTensorShapesFromInputs(), SpaceToDepthLayer::ValidateTensorShapesFromInputs(), Convolution2dLayer::ValidateTensorShapesFromInputs(), FullyConnectedLayer::ValidateTensorShapesFromInputs(), UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs(), LstmLayer::ValidateTensorShapesFromInputs(), BatchNormalizationLayer::ValidateTensorShapesFromInputs(), OutputLayer::ValidateTensorShapesFromInputs(), QuantizedLstmLayer::ValidateTensorShapesFromInputs(), and QLstmLayer::ValidateTensorShapesFromInputs().
|
inlineoverridevirtual |
Implements IInputSlot.
Definition at line 200 of file Layer.hpp.
References InputSlot::GetConnectedOutputSlot().
|
inline |
Definition at line 52 of file Layer.hpp.
Referenced by Graph::AddCompatibilityLayers(), OutputSlot::CalculateIndexOnOwner(), SplitterLayer::CreateWorkload(), armnn::optimizations::pad_fold::FoldPadIntoLayer2dImpl(), OutputSlot::GetOwningLayerGuid(), Graph::InsertNewLayer(), OutputSlot::IsTensorInfoSet(), OptimizeConsecutiveReshapesImpl::Run(), PermuteAndBatchToSpaceAsDepthToSpaceImpl< PermuteType >::Run(), OptimizeInverseConversionsImpl::Run(), OptimizeInversePermutesImpl< PermuteType >::Run(), MovePermuteUpImpl::Run(), MoveTransposeUpImpl::Run(), SquashEqualSiblingsImpl< Comparable >::Run(), FuseBatchNorm< ConvLayer, ArmnnType, T >::Run(), FoldPadIntoConvolution2dImpl::Run(), FoldPadIntoDepthwiseConvolution2dImpl::Run(), and SubgraphViewSelector::SelectSubgraphs().
|
inline |
Definition at line 53 of file Layer.hpp.
Referenced by Graph::AddCompatibilityLayers(), and SubgraphViewSelector::SelectSubgraphs().
void Insert | ( | Layer & | layer | ) |
Definition at line 20 of file Layer.cpp.
References ARMNN_ASSERT, OutputSlot::Connect(), InputSlot::GetConnectedOutputSlot(), Layer::GetInputSlot(), Layer::GetNumInputSlots(), Layer::GetNumOutputSlots(), Layer::GetOutputHandler(), Layer::GetOutputSlot(), OutputSlot::SetEdgeStrategy(), OutputHandler::SetTensorInfo(), and armnn::Undefined.
Referenced by Graph::InsertNewLayer(), and InputSlot::SetConnection().
|
inline |
Links the slot to an output slot or breaks an existing link if passing nullptr.
Definition at line 59 of file Layer.hpp.
References InputSlot::GetConnection(), and InputSlot::Insert().
Referenced by OutputSlot::Connect(), and OutputSlot::Disconnect().