ArmNN
NotReleased
|
#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 BOOST_AUTO_TEST_CASE(), ConcatLayer::CreateWorkload(), DebugLayer::CreateWorkload(), InputSlot::GetConnection(), InputSlot::Insert(), armnn::InsertConvertFp16ToFp32LayersBefore(), armnn::InsertConvertFp32ToFp16LayersAfter(), armnn::InsertDebugLayerAfter(), Graph::InsertNewLayer(), QuantizerVisitor::QuantizerVisitor(), FoldPadIntoConvolution2dImpl::Run(), OptimizeConsecutiveReshapesImpl::Run(), PermuteAndBatchToSpaceAsDepthToSpaceImpl::Run(), OptimizeInversePermutesImpl::Run(), OptimizeInverseConversionsImpl::Run(), MovePermuteUpImpl::Run(), and SquashEqualSiblingsImpl< Comparable >::Run().
|
inline |
|
inlineoverridevirtual |
Implements IInputSlot.
Definition at line 199 of file Layer.hpp.
References InputSlot::GetConnectedOutputSlot().
Referenced by BOOST_AUTO_TEST_CASE(), OutputSlot::DisconnectAll(), Layer::GetDataType(), OutputSlot::GetNumConnections(), IWorkloadFactory::IsLayerSupported(), OutputSlot::MoveAllConnections(), OutputSlot::operator==(), PermuteAndBatchToSpaceAsDepthToSpaceImpl::Run(), DepthwiseConvolution2dLayer::SerializeLayerParameters(), Convolution2dLayer::SerializeLayerParameters(), InputSlot::SetConnection(), Graph::SubstituteSubgraph(), ElementwiseBaseLayer::ValidateTensorShapesFromInputs(), QuantizeLayer::ValidateTensorShapesFromInputs(), ActivationLayer::ValidateTensorShapesFromInputs(), AbsLayer::ValidateTensorShapesFromInputs(), MemCopyLayer::ValidateTensorShapesFromInputs(), MemImportLayer::ValidateTensorShapesFromInputs(), DebugLayer::ValidateTensorShapesFromInputs(), MergeLayer::ValidateTensorShapesFromInputs(), NormalizationLayer::ValidateTensorShapesFromInputs(), PermuteLayer::ValidateTensorShapesFromInputs(), SliceLayer::ValidateTensorShapesFromInputs(), Pooling2dLayer::ValidateTensorShapesFromInputs(), ResizeLayer::ValidateTensorShapesFromInputs(), RsqrtLayer::ValidateTensorShapesFromInputs(), DequantizeLayer::ValidateTensorShapesFromInputs(), L2NormalizationLayer::ValidateTensorShapesFromInputs(), SoftmaxLayer::ValidateTensorShapesFromInputs(), StackLayer::ValidateTensorShapesFromInputs(), FloorLayer::ValidateTensorShapesFromInputs(), BatchToSpaceNdLayer::ValidateTensorShapesFromInputs(), FakeQuantizationLayer::ValidateTensorShapesFromInputs(), SwitchLayer::ValidateTensorShapesFromInputs(), InstanceNormalizationLayer::ValidateTensorShapesFromInputs(), ConvertFp32ToFp16Layer::ValidateTensorShapesFromInputs(), LogSoftmaxLayer::ValidateTensorShapesFromInputs(), MeanLayer::ValidateTensorShapesFromInputs(), GatherLayer::ValidateTensorShapesFromInputs(), ConvertFp16ToFp32Layer::ValidateTensorShapesFromInputs(), ArgMinMaxLayer::ValidateTensorShapesFromInputs(), StridedSliceLayer::ValidateTensorShapesFromInputs(), ElementwiseUnaryLayer::ValidateTensorShapesFromInputs(), SpaceToDepthLayer::ValidateTensorShapesFromInputs(), DepthToSpaceLayer::ValidateTensorShapesFromInputs(), PreluLayer::ValidateTensorShapesFromInputs(), SpaceToBatchNdLayer::ValidateTensorShapesFromInputs(), DepthwiseConvolution2dLayer::ValidateTensorShapesFromInputs(), TransposeConvolution2dLayer::ValidateTensorShapesFromInputs(), ComparisonLayer::ValidateTensorShapesFromInputs(), FullyConnectedLayer::ValidateTensorShapesFromInputs(), Convolution2dLayer::ValidateTensorShapesFromInputs(), BatchNormalizationLayer::ValidateTensorShapesFromInputs(), OutputLayer::ValidateTensorShapesFromInputs(), QuantizedLstmLayer::ValidateTensorShapesFromInputs(), and LstmLayer::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(), OutputSlot::GetOwningLayerGuid(), Graph::InsertNewLayer(), FoldPadIntoConvolution2dImpl::Run(), OptimizeConsecutiveReshapesImpl::Run(), PermuteAndBatchToSpaceAsDepthToSpaceImpl::Run(), OptimizeInversePermutesImpl::Run(), OptimizeInverseConversionsImpl::Run(), MovePermuteUpImpl::Run(), and SquashEqualSiblingsImpl< Comparable >::Run().
|
inline |
Definition at line 53 of file Layer.hpp.
Referenced by Graph::AddCompatibilityLayers().
void Insert | ( | Layer & | layer | ) |
Definition at line 20 of file Layer.cpp.
References 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().