InputSlot Class Reference

#include <Layer.hpp>

Inheritance diagram for InputSlot:

Public Member Functions
	InputSlot (Layer &owner, unsigned int slotIndex)
	~InputSlot ()
Layer &	GetOwningLayer () const
unsigned int	GetSlotIndex () const override
const OutputSlot *	GetConnectedOutputSlot () const
OutputSlot *	GetConnectedOutputSlot ()
const IConnectableLayer &	GetOwningIConnectableLayer () const override
IConnectableLayer &	GetOwningIConnectableLayer () override
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...

Detailed Description

Definition at line 42 of file Layer.hpp.

Constructor & Destructor Documentation

InputSlot()

InputSlot ( Layer &  owner, unsigned int  slotIndex  )

inlineexplicit

Definition at line 45 of file Layer.hpp.

References InputSlot::~InputSlot().

    : m_OwningLayer(owner)
    , m_Connection(nullptr)
    , m_SlotIndex(slotIndex)
    {}

~InputSlot()

~InputSlot ( )

inline

Definition at line 186 of file Layer.hpp.

Referenced by InputSlot::InputSlot().

{
    if (m_Connection != nullptr)
    {
        try
        {
            // Coverity fix: Disconnect() may throw uncaught exceptions.
            m_Connection->Disconnect(*this);
        }
        catch (const std::exception& e)
        {
            // Coverity fix: BOOST_LOG_TRIVIAL (typically used to report errors) may throw an
            // exception of type std::length_error.
            // Using stderr instead in this context as there is no point in nesting try-catch blocks here.
            std::cerr << "WARNING: An error has occurred when disconnecting an input slot: "
                      << e.what() << std::endl;
        }
    }
}

Member Function Documentation

GetConnectedOutputSlot() [1/2]

const OutputSlot* GetConnectedOutputSlot ( ) const

inline

Definition at line 56 of file Layer.hpp.

Referenced by ConcatLayer::CreateWorkload(), DebugLayer::CreateWorkload(), armnn::optimizations::pad_fold::FoldPadIntoLayer2dImpl(), LayerWithParameters< Pooling3dDescriptor >::GetConnectedConstantAsInputTensors(), InputSlot::GetConnection(), LoadedNetwork::ImportOutputs(), Graph::InferTensorInfos(), InputSlot::Insert(), armnn::InsertConvertBf16ToFp32LayersBefore(), armnn::InsertConvertFp16ToFp32LayersBefore(), armnn::InsertConvertFp32ToBf16LayersAfter(), armnn::InsertConvertFp32ToBf16LayersBefore(), armnn::InsertConvertFp32ToFp16LayersAfter(), armnn::InsertDebugLayerAfter(), Graph::InsertNewLayer(), RefBackend::OptimizeSubgraphView(), NeonBackend::OptimizeSubgraphView(), ClBackend::OptimizeSubgraphView(), LoadedNetwork::RegisterDebugCallback(), OptimizeConsecutiveReshapesImpl::Run(), ConvertConstPermuteLayersToConstLayers::Run(), FuseConvertFp32ToBf16IntoConstLayers::Run(), PermuteAndBatchToSpaceAsDepthToSpaceImpl< PermuteType >::Run(), ConvertConstDequantisationLayersToConstLayersImpl::Run(), OptimizeInverseConversionsImpl::Run(), MovePermuteUpImpl::Run(), PermuteDepthwiseConv2dWeightsImpl::Run(), MoveTransposeUpImpl::Run(), OptimizeInversePermutesImpl< PermuteType >::Run(), SquashEqualSiblingsImpl< Comparable >::Run(), AddBroadcastReshapeLayerImpl::Run(), and FuseBatchNorm< ConvLayer, ArmnnType, T >::Run().

{ return m_Connection; }

GetConnectedOutputSlot() [2/2]

OutputSlot* GetConnectedOutputSlot ( )

inline

Definition at line 57 of file Layer.hpp.

References InputSlot::GetOwningIConnectableLayer().

{ return m_Connection; }

GetConnection() [1/2]

const IOutputSlot * GetConnection ( ) const

inlineoverridevirtual

Implements IInputSlot.

Definition at line 206 of file Layer.hpp.

References InputSlot::GetConnectedOutputSlot().

Referenced by OutputSlot::DisconnectAll(), LayerWithParameters< Pooling3dDescriptor >::GetConnectedConstantAsInputTensors(), Layer::GetDataType(), OutputSlot::GetNumConnections(), OutputSlot::MoveAllConnections(), OutputSlot::operator==(), armnn::RevertConstantWeightsToFP32(), PermuteAndBatchToSpaceAsDepthToSpaceImpl< PermuteType >::Run(), Convolution3dLayer::SerializeLayerParameters(), DepthwiseConvolution2dLayer::SerializeLayerParameters(), Convolution2dLayer::SerializeLayerParameters(), InputSlot::SetConnection(), ElementwiseBaseLayer::ValidateTensorShapesFromInputs(), QuantizeLayer::ValidateTensorShapesFromInputs(), ActivationLayer::ValidateTensorShapesFromInputs(), ChannelShuffleLayer::ValidateTensorShapesFromInputs(), FillLayer::ValidateTensorShapesFromInputs(), ConvertFp32ToFp16Layer::ValidateTensorShapesFromInputs(), ReduceLayer::ValidateTensorShapesFromInputs(), SoftmaxLayer::ValidateTensorShapesFromInputs(), ConvertBf16ToFp32Layer::ValidateTensorShapesFromInputs(), FloorLayer::ValidateTensorShapesFromInputs(), ConvertFp16ToFp32Layer::ValidateTensorShapesFromInputs(), SwitchLayer::ValidateTensorShapesFromInputs(), InstanceNormalizationLayer::ValidateTensorShapesFromInputs(), L2NormalizationLayer::ValidateTensorShapesFromInputs(), ConvertFp32ToBf16Layer::ValidateTensorShapesFromInputs(), NormalizationLayer::ValidateTensorShapesFromInputs(), StackLayer::ValidateTensorShapesFromInputs(), MemCopyLayer::ValidateTensorShapesFromInputs(), MemImportLayer::ValidateTensorShapesFromInputs(), MergeLayer::ValidateTensorShapesFromInputs(), BatchToSpaceNdLayer::ValidateTensorShapesFromInputs(), DebugLayer::ValidateTensorShapesFromInputs(), Pooling2dLayer::ValidateTensorShapesFromInputs(), CastLayer::ValidateTensorShapesFromInputs(), Pooling3dLayer::ValidateTensorShapesFromInputs(), ResizeLayer::ValidateTensorShapesFromInputs(), ShapeLayer::ValidateTensorShapesFromInputs(), DequantizeLayer::ValidateTensorShapesFromInputs(), AbsLayer::ValidateTensorShapesFromInputs(), FakeQuantizationLayer::ValidateTensorShapesFromInputs(), RsqrtLayer::ValidateTensorShapesFromInputs(), SliceLayer::ValidateTensorShapesFromInputs(), TransposeLayer::ValidateTensorShapesFromInputs(), LogSoftmaxLayer::ValidateTensorShapesFromInputs(), MeanLayer::ValidateTensorShapesFromInputs(), PadLayer::ValidateTensorShapesFromInputs(), ReshapeLayer::ValidateTensorShapesFromInputs(), PermuteLayer::ValidateTensorShapesFromInputs(), Convolution3dLayer::ValidateTensorShapesFromInputs(), GatherNdLayer::ValidateTensorShapesFromInputs(), BatchMatMulLayer::ValidateTensorShapesFromInputs(), ArgMinMaxLayer::ValidateTensorShapesFromInputs(), ElementwiseUnaryLayer::ValidateTensorShapesFromInputs(), StridedSliceLayer::ValidateTensorShapesFromInputs(), GatherLayer::ValidateTensorShapesFromInputs(), TransposeConvolution2dLayer::ValidateTensorShapesFromInputs(), SpaceToBatchNdLayer::ValidateTensorShapesFromInputs(), SpaceToDepthLayer::ValidateTensorShapesFromInputs(), PreluLayer::ValidateTensorShapesFromInputs(), ComparisonLayer::ValidateTensorShapesFromInputs(), LogicalBinaryLayer::ValidateTensorShapesFromInputs(), DepthToSpaceLayer::ValidateTensorShapesFromInputs(), FullyConnectedLayer::ValidateTensorShapesFromInputs(), DepthwiseConvolution2dLayer::ValidateTensorShapesFromInputs(), LstmLayer::ValidateTensorShapesFromInputs(), Convolution2dLayer::ValidateTensorShapesFromInputs(), UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs(), BatchNormalizationLayer::ValidateTensorShapesFromInputs(), OutputLayer::ValidateTensorShapesFromInputs(), QuantizedLstmLayer::ValidateTensorShapesFromInputs(), and QLstmLayer::ValidateTensorShapesFromInputs().

{ return GetConnectedOutputSlot(); }

GetConnection() [2/2]

IOutputSlot * GetConnection ( )

inlineoverridevirtual

Implements IInputSlot.

Definition at line 207 of file Layer.hpp.

References InputSlot::GetConnectedOutputSlot().

{ return GetConnectedOutputSlot(); }

GetOwningIConnectableLayer() [1/2]

const IConnectableLayer & GetOwningIConnectableLayer ( ) const

overridevirtual

Implements IInputSlot.

Definition at line 551 of file Layer.cpp.

Referenced by InputSlot::GetConnectedOutputSlot(), and OutputSlot::GetOwningLayer().

{
    return m_OwningLayer;
}

GetOwningIConnectableLayer() [2/2]

IConnectableLayer & GetOwningIConnectableLayer ( )

overridevirtual

Implements IInputSlot.

Definition at line 556 of file Layer.cpp.

{
    return m_OwningLayer;
}

GetOwningLayer()

Layer& GetOwningLayer ( ) const

inline

Definition at line 53 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(), ConvertConstPermuteLayersToConstLayers::Run(), FuseConvertFp32ToBf16IntoConstLayers::Run(), OptimizeInverseConversionsImpl::Run(), PermuteAndBatchToSpaceAsDepthToSpaceImpl< PermuteType >::Run(), ConvertConstDequantisationLayersToConstLayersImpl::Run(), MovePermuteUpImpl::Run(), OptimizeInversePermutesImpl< PermuteType >::Run(), MoveTransposeUpImpl::Run(), SquashEqualSiblingsImpl< Comparable >::Run(), FuseBatchNorm< ConvLayer, ArmnnType, T >::Run(), FoldPadIntoConvolution2dImpl::Run(), and FoldPadIntoDepthwiseConvolution2dImpl::Run().

{ return m_OwningLayer; }

GetSlotIndex()

unsigned int GetSlotIndex ( ) const

inlineoverridevirtual

Implements IInputSlot.

Definition at line 54 of file Layer.hpp.

Referenced by Graph::AddCompatibilityLayers().

{ return m_SlotIndex; }

Insert()

void Insert

(

Layer &

layer

)

Definition at line 48 of file Layer.cpp.

References ARMNN_ASSERT, armnn::AssertNumberOfInputSlots(), OutputSlot::Connect(), InputSlot::GetConnectedOutputSlot(), Layer::GetInputSlot(), Layer::GetNumOutputSlots(), Layer::GetOutputHandler(), Layer::GetOutputSlot(), OutputSlot::SetEdgeStrategy(), OutputHandler::SetTensorInfo(), and armnn::Undefined.

Referenced by Graph::InsertNewLayer(), and InputSlot::SetConnection().

{
    ARMNN_ASSERT(layer.GetNumOutputSlots() == 1);

    OutputSlot* const prevSlot = GetConnectedOutputSlot();

    if (prevSlot != nullptr)
    {
        // Disconnects parent from this.
        prevSlot->Disconnect(*this);

        AssertNumberOfInputSlots(layer);

        // Connects inserted layer to parent.
        int idx = prevSlot->Connect(layer.GetInputSlot(0));
        prevSlot->SetEdgeStrategy(armnn::numeric_cast<unsigned int>(idx), EdgeStrategy::Undefined);

        // Sets tensor info for inserted layer.
        const TensorInfo& tensorInfo = prevSlot->GetTensorInfo();
        layer.GetOutputHandler().SetTensorInfo(tensorInfo);
    }

    // Connects inserted layer to this.
    layer.GetOutputSlot(0).Connect(*this);
    layer.GetOutputSlot(0).SetEdgeStrategy(0, EdgeStrategy::Undefined);
}

SetConnection()

void SetConnection ( OutputSlot *  source )

inline

Links the slot to an output slot or breaks an existing link if passing nullptr.

Definition at line 63 of file Layer.hpp.

References InputSlot::GetConnection(), and InputSlot::Insert().

Referenced by OutputSlot::Connect(), and OutputSlot::Disconnect().

    {
        if (m_Connection != nullptr && source != nullptr)
        {
            throw InvalidArgumentException("Tried to connect an output slot to an input slot, "
                "but the latter already has a connection");
        }
        m_Connection = source;
    }

---

The documentation for this class was generated from the following files:

• src/armnn/Layer.hpp
• src/armnn/Layer.cpp