OutputSlot Class Reference

#include <Layer.hpp>

Inheritance diagram for OutputSlot:

Public Member Functions
	OutputSlot (Layer &owner, OutputHandler &outputHandler)
	OutputSlot (const OutputSlot &)=delete
OutputSlot &	operator= (const OutputSlot &)=delete
OutputSlot &	operator= (OutputSlot &&)=delete
	OutputSlot (OutputSlot &&)=default
	~OutputSlot ()
Layer &	GetOwningLayer () const
const IConnectableLayer &	GetOwningIConnectableLayer () const override
IConnectableLayer &	GetOwningIConnectableLayer () override
LayerGuid	GetOwningLayerGuid () const override
const OutputHandler &	GetOutputHandler () const
OutputHandler &	GetOutputHandler ()
int	Connect (InputSlot &destination)
void	Disconnect (InputSlot &slot)
const std::vector< InputSlot * > &	GetConnections () const
const std::vector< EdgeStrategy > &	GetEdgeStrategies () const
bool	ValidateTensorShape (const TensorShape &shape) const
void	DisconnectAll ()
void	MoveAllConnections (OutputSlot &destination)
	Moves all connections to another OutputSlot. More...
unsigned int	GetNumConnections () const override
const InputSlot *	GetConnection (unsigned int index) const override
InputSlot *	GetConnection (unsigned int index) override
void	SetTensorInfo (const TensorInfo &tensorInfo) override
const TensorInfo &	GetTensorInfo () const override
bool	IsTensorInfoSet () const override
int	Connect (IInputSlot &destination) override
void	Disconnect (IInputSlot &slot) override
unsigned int	CalculateIndexOnOwner () const override
bool	operator== (const OutputSlot &other) const
void	SetTensorHandleFactory (const ITensorHandleFactory::FactoryId &id)
ITensorHandleFactory::FactoryId	GetTensorHandleFactoryId () const
void	SetEdgeStrategy (unsigned int connectionIndex, EdgeStrategy strategy)
EdgeStrategy	GetEdgeStrategyForConnection (unsigned int connectionIdx) const

Additional Inherited Members
Protected Member Functions inherited from IOutputSlot
	~IOutputSlot ()
	Not user deletable. More...

Detailed Description

Definition at line 87 of file Layer.hpp.

Constructor & Destructor Documentation

OutputSlot() [1/3]

OutputSlot ( Layer &  owner, OutputHandler &  outputHandler  )

inlineexplicit

Definition at line 90 of file Layer.hpp.

    : m_OwningLayer(owner)
    , m_OutputHandler(outputHandler)
    , m_TensorHandleFactoryId(ITensorHandleFactory::LegacyFactoryId)
    {}

OutputSlot() [2/3]

OutputSlot ( const OutputSlot &  )

delete

OutputSlot() [3/3]

OutputSlot ( OutputSlot &&  )

default

~OutputSlot()

~OutputSlot ( )

inline

Definition at line 102 of file Layer.hpp.

    {
        try
        {
            // Coverity fix: DisconnectAll() may throw uncaught exceptions.
            DisconnectAll();
        }
        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 all output slots: "
                      << e.what() << std::endl;
        }
    }

References OutputSlot::DisconnectAll().

Member Function Documentation

CalculateIndexOnOwner()

unsigned int CalculateIndexOnOwner ( ) const

overridevirtual

Implements IOutputSlot.

Definition at line 159 of file Layer.cpp.

{
    for (unsigned int i = 0; i < GetOwningLayer().GetNumOutputSlots(); i++)
    {
        if (GetOwningLayer().GetOutputSlot(i) == (*this))
        {
            return i;
        }
    }
    ARMNN_ASSERT_MSG(false, "Did not find slot on owner.");
    return 0; // Error
}

References ARMNN_ASSERT_MSG, Layer::GetNumOutputSlots(), Layer::GetOutputSlot(), and OutputSlot::GetOwningLayer().

Referenced by DebugLayer::CreateWorkload().

Connect() [1/2]

int Connect ( IInputSlot &  destination )

inlineoverridevirtual

Implements IOutputSlot.

Definition at line 153 of file Layer.hpp.

    {
        return Connect(*PolymorphicDowncast<InputSlot*>(&destination));
    }

References OutputSlot::Connect().

Connect() [2/2]

int Connect

(

InputSlot &

destination

)

Examples

AsyncExecutionSample.cpp, and SimpleSample.cpp.

Definition at line 112 of file Layer.cpp.

{
    destination.SetConnection(this);
    m_Connections.push_back(&destination);
    m_EdgeStrategies.push_back(EdgeStrategy::Undefined);
    return armnn::numeric_cast<int>(m_Connections.size() - 1);
}

References InputSlot::SetConnection(), and armnn::Undefined.

Referenced by OutputSlot::Connect(), armnn::optimizations::pad_fold::FoldPadIntoLayer2dImpl(), InputSlot::Insert(), Graph::InsertNewLayer(), OutputSlot::MoveAllConnections(), and FuseBatchNorm< ConvLayer, ArmnnType, T >::Run().

Disconnect() [1/2]

void Disconnect ( IInputSlot &  slot )

inlineoverridevirtual

Implements IOutputSlot.

Definition at line 158 of file Layer.hpp.

    {
        return Disconnect(*PolymorphicDowncast<InputSlot*>(&slot));
    }

References OutputSlot::Disconnect().

Disconnect() [2/2]

void Disconnect

(

InputSlot &

slot

)

Definition at line 120 of file Layer.cpp.

{
    slot.SetConnection(nullptr);
    auto it = std::find(m_Connections.begin(), m_Connections.end(), &slot);
 
    if (it == m_Connections.end())
    {
        return;
    }
 
    auto idx = std::distance(m_Connections.begin(), it);
    m_Connections.erase(std::remove(m_Connections.begin(), m_Connections.end(), &slot), m_Connections.end());
 
    m_EdgeStrategies.erase(m_EdgeStrategies.begin() + idx);
}

References InputSlot::SetConnection().

Referenced by OutputSlot::Disconnect(), OutputSlot::DisconnectAll(), armnn::optimizations::pad_fold::FoldPadIntoLayer2dImpl(), InputSlot::Insert(), OutputSlot::MoveAllConnections(), FuseBatchNorm< ConvLayer, ArmnnType, T >::Run(), and InputSlot::~InputSlot().

DisconnectAll()

void DisconnectAll

(

)

Definition at line 136 of file Layer.cpp.

{
    while (GetNumConnections() > 0)
    {
        InputSlot& connection = *GetConnection(0);
        Disconnect(connection);
    }
}

References OutputSlot::Disconnect(), OutputSlot::GetConnection(), and OutputSlot::GetNumConnections().

Referenced by OutputSlot::~OutputSlot().

GetConnection() [1/2]

const InputSlot * GetConnection ( unsigned int  index ) const

overridevirtual

Implements IOutputSlot.

Definition at line 75 of file Layer.cpp.

{
    ValidateConnectionIndex(index);
    return m_Connections[index];
}

Referenced by OutputSlot::DisconnectAll(), GenerateUniqueOutputName(), OutputSlot::MoveAllConnections(), and OutputSlot::operator==().

GetConnection() [2/2]

InputSlot * GetConnection ( unsigned int  index )

overridevirtual

Implements IOutputSlot.

Definition at line 81 of file Layer.cpp.

{
    ValidateConnectionIndex(index);
    return m_Connections[index];
}

GetConnections()

const std::vector<InputSlot*>& GetConnections ( ) const

inline

Definition at line 132 of file Layer.hpp.

{ return m_Connections; }

Referenced by Graph::AddCompatibilityLayers(), armnn::CalculateSlotOption(), armnn::CalculateSlotOptionForInput(), SquashEqualSiblingsImpl< Comparable >::Run(), and armnn::SelectTensorHandleStrategy().

GetEdgeStrategies()

const std::vector<EdgeStrategy>& GetEdgeStrategies ( ) const

inline

Definition at line 133 of file Layer.hpp.

{ return m_EdgeStrategies; }

Referenced by Graph::AddCompatibilityLayers().

GetEdgeStrategyForConnection()

EdgeStrategy GetEdgeStrategyForConnection

(

unsigned int

connectionIdx

)

const

Definition at line 215 of file Layer.cpp.

{
    return m_EdgeStrategies[connectionIdx];
}

GetNumConnections()

unsigned int GetNumConnections ( ) const

inlineoverridevirtual

Implements IOutputSlot.

Definition at line 145 of file Layer.hpp.

{ return armnn::numeric_cast<unsigned int>(m_Connections.size()); }

Referenced by OutputSlot::DisconnectAll(), OutputSlot::MoveAllConnections(), OutputSlot::operator==(), RefBackend::OptimizeSubgraphView(), ClBackend::OptimizeSubgraphView(), OptimizeConsecutiveReshapesImpl::Run(), MovePermuteUpImpl::Run(), MoveTransposeUpImpl::Run(), SquashEqualSiblingsImpl< Comparable >::Run(), and AddBroadcastReshapeLayerImpl::Run().

GetOutputHandler() [1/2]

OutputHandler& GetOutputHandler ( )

inline

Definition at line 127 of file Layer.hpp.

{ return m_OutputHandler; }

GetOutputHandler() [2/2]

const OutputHandler& GetOutputHandler ( ) const

inline

Definition at line 126 of file Layer.hpp.

{ return m_OutputHandler; }

Referenced by OutputSlot::GetTensorInfo(), OutputSlot::IsTensorInfoSet(), OutputSlot::MoveAllConnections(), and OutputSlot::SetTensorInfo().

GetOwningIConnectableLayer() [1/2]

IConnectableLayer & GetOwningIConnectableLayer ( ) const

overridevirtual

Implements IOutputSlot.

Definition at line 554 of file Layer.cpp.

{
    return m_OwningLayer;
}

GetOwningIConnectableLayer() [2/2]

IConnectableLayer& GetOwningIConnectableLayer ( )

overridevirtual

Implements IOutputSlot.

GetOwningLayer()

Layer& GetOwningLayer ( ) const

inline

Definition at line 119 of file Layer.hpp.

{ return m_OwningLayer; }

Referenced by OutputSlot::CalculateIndexOnOwner(), armnn::CalculateSlotOption(), armnn::CalculateSlotOptionForInput(), ConvertAvgPool2DIgnoreValueToTosaOperator(), ConvertConcatToTosaOperator(), ConvertConv2dToTosaOperator(), ConvertElementwiseBinaryToTosaOperator(), ConvertElementwiseUnaryOperator(), ConvertPooling2DToTosaOperator(), ConvertReshapeToTosaOperator(), ConvertSliceToTosaOperator(), ConvertTransposeConv2dToTosaOperator(), ConvertTransposeToTosaOperator(), DebugLayer::CreateWorkload(), armnn::optimizations::pad_fold::FoldPadIntoLayer2dImpl(), LayerWithParameters< StridedSliceDescriptor >::GetConnectedConstantAsInputTensors(), OutputSlot::GetOwningLayerGuid(), Layer::GetPriority(), Graph::InsertNewLayer(), OutputSlot::IsTensorInfoSet(), RefBackend::OptimizeSubgraphView(), ClBackend::OptimizeSubgraphView(), OptimizeConsecutiveReshapesImpl::Run(), ConvertConstPermuteLayersToConstLayers::Run(), OptimizeInverseConversionsImpl::Run(), PermuteAndBatchToSpaceAsDepthToSpaceImpl< PermuteType >::Run(), ConvertConstDequantisationLayersToConstLayersImpl::Run(), MovePermuteUpImpl::Run(), MoveTransposeUpImpl::Run(), OptimizeInversePermutesImpl< PermuteType >::Run(), FuseBatchNorm< ConvLayer, ArmnnType, T >::Run(), AddBroadcastReshapeLayerImpl::Run(), and Graph::SerializeToDot().

GetOwningLayerGuid()

LayerGuid GetOwningLayerGuid ( ) const

overridevirtual

Implements IOutputSlot.

Definition at line 195 of file Layer.cpp.

{
    return GetOwningLayer().GetGuid();
}

References Layer::GetGuid(), and OutputSlot::GetOwningLayer().

GetTensorHandleFactoryId()

ITensorHandleFactory::FactoryId GetTensorHandleFactoryId

(

)

const

Definition at line 205 of file Layer.cpp.

{
    return m_TensorHandleFactoryId;
}

Referenced by Graph::AddCompatibilityLayers(), SplitterLayer::CreateTensorHandles(), ConcatLayer::CreateTensorHandles(), Layer::CreateTensorHandles(), LoadedNetwork::ImportInputs(), and LoadedNetwork::ImportOutputs().

GetTensorInfo()

const TensorInfo & GetTensorInfo ( ) const

overridevirtual

Implements IOutputSlot.

Definition at line 92 of file Layer.cpp.

{
    return GetOutputHandler().GetTensorInfo();
}

References OutputSlot::GetOutputHandler(), and OutputHandler::GetTensorInfo().

Referenced by armnn::CheckScaleSetOnQuantizedType(), armnn::optimizations::pad_fold::FoldPadIntoLayer2dImpl(), Layer::GetDataType(), armnn::GetLayerInOutDatatype(), LoadedNetwork::ImportInputs(), LoadedNetwork::ImportOutputs(), InputSlot::Insert(), armnn::InsertConvertFp16ToFp32LayersBefore(), armnn::InsertConvertFp32ToFp16LayersAfter(), armnn::InsertDebugLayerAfter(), OutputSlot::MoveAllConnections(), RefBackend::OptimizeSubgraphView(), NeonBackend::OptimizeSubgraphView(), ClBackend::OptimizeSubgraphView(), OptimizeConsecutiveReshapesImpl::Run(), PermuteAndBatchToSpaceAsDepthToSpaceImpl< PermuteType >::Run(), AddBroadcastReshapeLayerImpl::Run(), FuseBatchNorm< ConvLayer, ArmnnType, T >::Run(), Graph::SerializeToDot(), Layer::ValidateAndCopyShape(), ElementwiseBaseLayer::ValidateTensorShapesFromInputs(), QuantizeLayer::ValidateTensorShapesFromInputs(), ChannelShuffleLayer::ValidateTensorShapesFromInputs(), ActivationLayer::ValidateTensorShapesFromInputs(), ConvertFp32ToFp16Layer::ValidateTensorShapesFromInputs(), RankLayer::ValidateTensorShapesFromInputs(), FillLayer::ValidateTensorShapesFromInputs(), SwitchLayer::ValidateTensorShapesFromInputs(), TransposeLayer::ValidateTensorShapesFromInputs(), ConvertFp16ToFp32Layer::ValidateTensorShapesFromInputs(), L2NormalizationLayer::ValidateTensorShapesFromInputs(), InstanceNormalizationLayer::ValidateTensorShapesFromInputs(), MemImportLayer::ValidateTensorShapesFromInputs(), MemCopyLayer::ValidateTensorShapesFromInputs(), DebugLayer::ValidateTensorShapesFromInputs(), MergeLayer::ValidateTensorShapesFromInputs(), NormalizationLayer::ValidateTensorShapesFromInputs(), Pooling3dLayer::ValidateTensorShapesFromInputs(), BatchToSpaceNdLayer::ValidateTensorShapesFromInputs(), PermuteLayer::ValidateTensorShapesFromInputs(), DequantizeLayer::ValidateTensorShapesFromInputs(), Pooling2dLayer::ValidateTensorShapesFromInputs(), SliceLayer::ValidateTensorShapesFromInputs(), CastLayer::ValidateTensorShapesFromInputs(), AbsLayer::ValidateTensorShapesFromInputs(), ResizeLayer::ValidateTensorShapesFromInputs(), SoftmaxLayer::ValidateTensorShapesFromInputs(), RsqrtLayer::ValidateTensorShapesFromInputs(), ShapeLayer::ValidateTensorShapesFromInputs(), FakeQuantizationLayer::ValidateTensorShapesFromInputs(), FloorLayer::ValidateTensorShapesFromInputs(), StackLayer::ValidateTensorShapesFromInputs(), LogSoftmaxLayer::ValidateTensorShapesFromInputs(), MeanLayer::ValidateTensorShapesFromInputs(), PadLayer::ValidateTensorShapesFromInputs(), ReshapeLayer::ValidateTensorShapesFromInputs(), Convolution2dLayer::ValidateTensorShapesFromInputs(), Convolution3dLayer::ValidateTensorShapesFromInputs(), DepthwiseConvolution2dLayer::ValidateTensorShapesFromInputs(), FullyConnectedLayer::ValidateTensorShapesFromInputs(), GatherNdLayer::ValidateTensorShapesFromInputs(), ElementwiseBinaryLayer::ValidateTensorShapesFromInputs(), BatchMatMulLayer::ValidateTensorShapesFromInputs(), GatherLayer::ValidateTensorShapesFromInputs(), ReduceLayer::ValidateTensorShapesFromInputs(), StridedSliceLayer::ValidateTensorShapesFromInputs(), DetectionPostProcessLayer::ValidateTensorShapesFromInputs(), ArgMinMaxLayer::ValidateTensorShapesFromInputs(), ElementwiseUnaryLayer::ValidateTensorShapesFromInputs(), TransposeConvolution2dLayer::ValidateTensorShapesFromInputs(), ComparisonLayer::ValidateTensorShapesFromInputs(), DepthToSpaceLayer::ValidateTensorShapesFromInputs(), SpaceToDepthLayer::ValidateTensorShapesFromInputs(), PreluLayer::ValidateTensorShapesFromInputs(), LogicalBinaryLayer::ValidateTensorShapesFromInputs(), SpaceToBatchNdLayer::ValidateTensorShapesFromInputs(), SplitterLayer::ValidateTensorShapesFromInputs(), UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs(), ConcatLayer::ValidateTensorShapesFromInputs(), LstmLayer::ValidateTensorShapesFromInputs(), BatchNormalizationLayer::ValidateTensorShapesFromInputs(), QuantizedLstmLayer::ValidateTensorShapesFromInputs(), and QLstmLayer::ValidateTensorShapesFromInputs().

IsTensorInfoSet()

bool IsTensorInfoSet ( ) const

overridevirtual

Implements IOutputSlot.

Definition at line 97 of file Layer.cpp.

{
    if (GetOwningLayer().GetShapeInferenceMethod() == ShapeInferenceMethod::InferAndValidate)
    {
        GetOwningLayer().ValidateTensorShapesFromInputs();
    }
    return GetOutputHandler().IsTensorInfoSet();
}

References OutputSlot::GetOutputHandler(), OutputSlot::GetOwningLayer(), armnn::InferAndValidate, OutputHandler::IsTensorInfoSet(), and Layer::ValidateTensorShapesFromInputs().

Referenced by AddBroadcastReshapeLayerImpl::Run(), and OutputSlot::ValidateTensorShape().

MoveAllConnections()

void MoveAllConnections

(

OutputSlot &

destination

)

Moves all connections to another OutputSlot.

Definition at line 145 of file Layer.cpp.

{
    while (GetNumConnections() > 0)
    {
        ARMNN_ASSERT_MSG(m_EdgeStrategies[0] == EdgeStrategy::Undefined,
            "Cannot move connections once memory strategies have be established.");
 
        InputSlot& connection = *GetConnection(0);
        Disconnect(connection);
        destination.Connect(connection);
        destination.GetOutputHandler().SetTensorInfo(GetOutputHandler().GetTensorInfo());
    }
}

References ARMNN_ASSERT_MSG, OutputSlot::Connect(), OutputSlot::Disconnect(), OutputSlot::GetConnection(), OutputSlot::GetNumConnections(), OutputSlot::GetOutputHandler(), OutputSlot::GetTensorInfo(), OutputHandler::SetTensorInfo(), and armnn::Undefined.

Referenced by Graph::InsertNewLayer(), PermuteAsReshapeImpl::Run(), TransposeAsReshapeImpl::Run(), OptimizeConsecutiveReshapesImpl::Run(), OptimizeInverseConversionsImpl::Run(), and FuseBatchNorm< ConvLayer, ArmnnType, T >::Run().

operator=() [1/2]

OutputSlot& operator= ( const OutputSlot &  )

delete

operator=() [2/2]

OutputSlot& operator= ( OutputSlot &&  )

delete

operator==()

bool operator==

(

const OutputSlot &

other

)

const

Definition at line 172 of file Layer.cpp.

{
    bool isSame = other.GetNumConnections() == GetNumConnections();
    if (!isSame)
    {
        return false;
    }
 
    for (unsigned int i = 0; i < GetNumConnections(); i++)
    {
        isSame &= other.GetConnection(i) == GetConnection(i);
    }
    return isSame;
}

References OutputSlot::GetConnection(), and OutputSlot::GetNumConnections().

SetEdgeStrategy()

void SetEdgeStrategy	(	unsigned int	connectionIndex,
		EdgeStrategy	strategy
	)

Definition at line 210 of file Layer.cpp.

{
    m_EdgeStrategies[connectionIndex] = strategy;
}

Referenced by Graph::AddCompatibilityLayers(), InputSlot::Insert(), and armnn::SelectTensorHandleStrategy().

SetTensorHandleFactory()

void SetTensorHandleFactory

(

const ITensorHandleFactory::FactoryId &

id

)

Definition at line 200 of file Layer.cpp.

{
    m_TensorHandleFactoryId = id;
}

Referenced by Graph::AddCompatibilityLayers(), and armnn::SelectTensorHandleStrategy().

SetTensorInfo()

void SetTensorInfo ( const TensorInfo &  tensorInfo )

overridevirtual

Implements IOutputSlot.

Examples

AsyncExecutionSample.cpp, and SimpleSample.cpp.

Definition at line 87 of file Layer.cpp.

{
    GetOutputHandler().SetTensorInfo(tensorInfo);
}

References OutputSlot::GetOutputHandler(), and OutputHandler::SetTensorInfo().

Referenced by armnn::AttemptBackendAssignment(), armnn::ChainReduceLayers(), armnn::CheckScaleSetOnQuantizedType(), armnn::InsertConvertFp16ToFp32LayersBefore(), armnn::InsertConvertFp32ToFp16LayersAfter(), armnn::InsertDebugLayerAfter(), FuseBatchNorm< ConvLayer, ArmnnType, T >::Run(), AddBroadcastReshapeLayerImpl::Run(), and Layer::ValidateAndCopyShape().

ValidateTensorShape()

bool ValidateTensorShape

(

const TensorShape &

shape

)

const

Definition at line 106 of file Layer.cpp.

{
    ARMNN_ASSERT_MSG(IsTensorInfoSet(), "TensorInfo must be set in order to validate the shape.");
    return shape == m_OutputHandler.GetTensorInfo().GetShape();
}

References ARMNN_ASSERT_MSG, TensorInfo::GetShape(), OutputHandler::GetTensorInfo(), and OutputSlot::IsTensorInfoSet().

---

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

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