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
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...

Detailed Description

Definition at line 41 of file Layer.hpp.

Constructor & Destructor Documentation

InputSlot()

InputSlot ( Layer &  owner, unsigned int  slotIndex  )

inlineexplicit

Definition at line 44 of file Layer.hpp.

References InputSlot::~InputSlot().

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

~InputSlot()

~InputSlot ( )

inline

Definition at line 179 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 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< PermuteType >::Run(), OptimizeInverseConversionsImpl::Run(), MoveTransposeUpImpl::Run(), MovePermuteUpImpl::Run(), OptimizeInversePermutesImpl< PermuteType >::Run(), and SquashEqualSiblingsImpl< Comparable >::Run().

{ return m_Connection; }

GetConnectedOutputSlot() [2/2]

OutputSlot* GetConnectedOutputSlot ( )

inline

Definition at line 56 of file Layer.hpp.

{ return m_Connection; }

GetConnection() [1/2]

const IOutputSlot * GetConnection ( ) const

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< PermuteType >::Run(), DepthwiseConvolution2dLayer::SerializeLayerParameters(), Convolution2dLayer::SerializeLayerParameters(), InputSlot::SetConnection(), Graph::SubstituteSubgraph(), ElementwiseBaseLayer::ValidateTensorShapesFromInputs(), QuantizeLayer::ValidateTensorShapesFromInputs(), ActivationLayer::ValidateTensorShapesFromInputs(), TransposeLayer::ValidateTensorShapesFromInputs(), StackLayer::ValidateTensorShapesFromInputs(), AbsLayer::ValidateTensorShapesFromInputs(), MemCopyLayer::ValidateTensorShapesFromInputs(), MemImportLayer::ValidateTensorShapesFromInputs(), DebugLayer::ValidateTensorShapesFromInputs(), MergeLayer::ValidateTensorShapesFromInputs(), NormalizationLayer::ValidateTensorShapesFromInputs(), PermuteLayer::ValidateTensorShapesFromInputs(), SliceLayer::ValidateTensorShapesFromInputs(), SoftmaxLayer::ValidateTensorShapesFromInputs(), Pooling2dLayer::ValidateTensorShapesFromInputs(), ResizeLayer::ValidateTensorShapesFromInputs(), L2NormalizationLayer::ValidateTensorShapesFromInputs(), RsqrtLayer::ValidateTensorShapesFromInputs(), DequantizeLayer::ValidateTensorShapesFromInputs(), BatchToSpaceNdLayer::ValidateTensorShapesFromInputs(), FakeQuantizationLayer::ValidateTensorShapesFromInputs(), SwitchLayer::ValidateTensorShapesFromInputs(), FloorLayer::ValidateTensorShapesFromInputs(), InstanceNormalizationLayer::ValidateTensorShapesFromInputs(), ConvertFp32ToFp16Layer::ValidateTensorShapesFromInputs(), MeanLayer::ValidateTensorShapesFromInputs(), ConvertFp16ToFp32Layer::ValidateTensorShapesFromInputs(), GatherLayer::ValidateTensorShapesFromInputs(), LogSoftmaxLayer::ValidateTensorShapesFromInputs(), ArgMinMaxLayer::ValidateTensorShapesFromInputs(), ElementwiseUnaryLayer::ValidateTensorShapesFromInputs(), StridedSliceLayer::ValidateTensorShapesFromInputs(), DepthToSpaceLayer::ValidateTensorShapesFromInputs(), ComparisonLayer::ValidateTensorShapesFromInputs(), SpaceToDepthLayer::ValidateTensorShapesFromInputs(), PreluLayer::ValidateTensorShapesFromInputs(), FullyConnectedLayer::ValidateTensorShapesFromInputs(), DepthwiseConvolution2dLayer::ValidateTensorShapesFromInputs(), TransposeConvolution2dLayer::ValidateTensorShapesFromInputs(), SpaceToBatchNdLayer::ValidateTensorShapesFromInputs(), Convolution2dLayer::ValidateTensorShapesFromInputs(), BatchNormalizationLayer::ValidateTensorShapesFromInputs(), OutputLayer::ValidateTensorShapesFromInputs(), QuantizedLstmLayer::ValidateTensorShapesFromInputs(), and LstmLayer::ValidateTensorShapesFromInputs().

{ return GetConnectedOutputSlot(); }

GetConnection() [2/2]

IOutputSlot * GetConnection ( )

inlineoverridevirtual

Implements IInputSlot.

Definition at line 200 of file Layer.hpp.

References InputSlot::GetConnectedOutputSlot().

{ return GetConnectedOutputSlot(); }

GetOwningLayer()

Layer& GetOwningLayer ( ) const

inline

Definition at line 52 of file Layer.hpp.

Referenced by Graph::AddCompatibilityLayers(), OutputSlot::CalculateIndexOnOwner(), OutputSlot::GetOwningLayerGuid(), Graph::InsertNewLayer(), OptimizeConsecutiveReshapesImpl::Run(), FoldPadIntoConvolution2dImpl::Run(), PermuteAndBatchToSpaceAsDepthToSpaceImpl< PermuteType >::Run(), OptimizeInverseConversionsImpl::Run(), MoveTransposeUpImpl::Run(), OptimizeInversePermutesImpl< PermuteType >::Run(), MovePermuteUpImpl::Run(), and SquashEqualSiblingsImpl< Comparable >::Run().

{ return m_OwningLayer; }

GetSlotIndex()

unsigned int GetSlotIndex ( ) const

inline

Definition at line 53 of file Layer.hpp.

Referenced by Graph::AddCompatibilityLayers().

{ return m_SlotIndex; }

Insert()

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().

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

    OutputSlot* const prevSlot = GetConnectedOutputSlot();

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

        // Connects inserted layer to parent.
        BOOST_ASSERT(layer.GetNumInputSlots() == 1);
        int idx = prevSlot->Connect(layer.GetInputSlot(0));
        prevSlot->SetEdgeStrategy(boost::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 59 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