Interface for a layer that is connectable to other layers via InputSlots and OutputSlots. More...

#include <INetwork.hpp>

Inheritance diagram for IConnectableLayer:

Public Types
using	ConstantTensors = std::vector< std::reference_wrapper< std::shared_ptr< ConstTensorHandle > >>

Public Member Functions
virtual const char *	GetName () const =0
	Returns the name of the layer. More...

virtual unsigned int	GetNumInputSlots () const =0
	Returns the number of connectable input slots. More...

virtual unsigned int	GetNumOutputSlots () const =0
	Returns the number of connectable output slots. More...

virtual const IInputSlot &	GetInputSlot (unsigned int index) const =0
	Get a const input slot handle by slot index. More...

virtual IInputSlot &	GetInputSlot (unsigned int index)=0
	Get the input slot handle by slot index. More...

virtual const IOutputSlot &	GetOutputSlot (unsigned int index) const =0
	Get the const output slot handle by slot index. More...

virtual IOutputSlot &	GetOutputSlot (unsigned int index)=0
	Get the output slot handle by slot index. More...

virtual std::vector< TensorShape >	InferOutputShapes (const std::vector< TensorShape > &inputShapes) const =0
	Infer the shape of the output(s) based on the provided input shape(s) More...

virtual LayerGuid	GetGuid () const =0
	Returns the unique id of the layer. More...

virtual void	ExecuteStrategy (IStrategy &strategy) const =0
	Apply a visitor to this layer. More...

virtual void	BackendSelectionHint (Optional< BackendId > backend)=0
	Provide a hint for the optimizer as to which backend to prefer for this layer. More...

virtual LayerType	GetType () const =0
	Returns the armnn::LayerType of this layer. More...

virtual const BaseDescriptor &	GetParameters () const =0
	If the layer has a descriptor return it. More...

virtual void	SetBackendId (const BackendId &id)=0
	Set the backend of the IConnectableLayer. More...

virtual ConstantTensors	GetConstantTensorsByRef ()=0

Protected Member Functions
	~IConnectableLayer ()
	Objects are not deletable via the handle. More...

Detailed Description

Interface for a layer that is connectable to other layers via InputSlots and OutputSlots.

Examples: AsyncExecutionSample.cpp, CustomMemoryAllocatorSample.cpp, DynamicSample.cpp, and SimpleSample.cpp.

Definition at line 68 of file INetwork.hpp.

Member Typedef Documentation

◆ ConstantTensors

using ConstantTensors = std::vector<std::reference_wrapper<std::shared_ptr<ConstTensorHandle> >>

Definition at line 124 of file INetwork.hpp.

Constructor & Destructor Documentation

◆ ~IConnectableLayer()

~IConnectableLayer ( )

inlineprotected

Objects are not deletable via the handle.

Definition at line 131 of file INetwork.hpp.

131 {}

Member Function Documentation

◆ BackendSelectionHint()

virtual void BackendSelectionHint ( Optional< BackendId > backend )

pure virtual

Provide a hint for the optimizer as to which backend to prefer for this layer.

By providing a BackendSelectionHint there is no guarantee the input backend supports that layer. If IsLayerSupported() returns false with the backend hint, we default to calling IsLayerSupported() on the BackendPreferences vector. Use SetBackendId() if we can guarantee a backend supports that layer (IsLayerSupported returns true for a specific backend).

Implemented in Layer.

◆ ExecuteStrategy()

virtual void ExecuteStrategy ( IStrategy & strategy ) const

pure virtual

Apply a visitor to this layer.

◆ GetConstantTensorsByRef()

virtual ConstantTensors GetConstantTensorsByRef ( )

pure virtual

Implemented in Layer, QLstmLayer, QuantizedLstmLayer, UnidirectionalSequenceLstmLayer, LstmLayer, Convolution2dLayer, DepthwiseConvolution2dLayer, FullyConnectedLayer, TransposeConvolution2dLayer, ConstantLayer, DetectionPostProcessLayer, and BatchNormalizationLayer.

◆ GetGuid()

virtual LayerGuid GetGuid ( ) const

pure virtual

Returns the unique id of the layer.

Implemented in Layer.

Referenced by RangeTracker::SetRange().

◆ GetInputSlot() [1/2]

virtual const IInputSlot& GetInputSlot ( unsigned int index ) const

pure virtual

Get a const input slot handle by slot index.

Implemented in Layer.

Examples: AsyncExecutionSample.cpp, CustomMemoryAllocatorSample.cpp, DynamicSample.cpp, and SimpleSample.cpp.

Referenced by Connect(), armnn_driver::ConvertPooling2d(), armnn_driver::ConvertReduce(), armnn_driver::ConvertToActivation(), armnn_driver::IsConnectedToDequantize(), armnn_driver::ProcessActivation(), and armnn_driver::SetupAndTrackLayerOutputSlot().

◆ GetInputSlot() [2/2]

virtual IInputSlot& GetInputSlot ( unsigned int index )

pure virtual

Get the input slot handle by slot index.

Implemented in Layer.

◆ GetName()

virtual const char* GetName ( ) const

pure virtual

Returns the name of the layer.

Implemented in Layer.

Referenced by Connect(), and SerializerStrategy::ExecuteStrategy().

◆ GetNumInputSlots()

virtual unsigned int GetNumInputSlots ( ) const

pure virtual

Returns the number of connectable input slots.

Implemented in Layer.

Referenced by armnn_driver::SetupAndTrackLayerOutputSlot().

◆ GetNumOutputSlots()

virtual unsigned int GetNumOutputSlots ( ) const

pure virtual

Returns the number of connectable output slots.

Implemented in Layer.

Referenced by SubgraphView::GetWorkingCopy(), armnn_driver::ProcessActivation(), RangeTracker::SetRange(), and armnn_driver::SetupAndTrackLayerOutputSlot().

◆ GetOutputSlot() [1/2]

virtual const IOutputSlot& GetOutputSlot ( unsigned int index ) const

pure virtual

Get the const output slot handle by slot index.

Implemented in Layer.

Examples: AsyncExecutionSample.cpp, CustomMemoryAllocatorSample.cpp, DynamicSample.cpp, and SimpleSample.cpp.

Referenced by Connect(), armnn_driver::ConvertToLayerInputHandle(), SubgraphView::GetWorkingCopy(), TosaRefBackend::OptimizeSubgraphView(), armnn_driver::ProcessActivation(), and armnn_driver::SetupAndTrackLayerOutputSlot().

◆ GetOutputSlot() [2/2]

virtual IOutputSlot& GetOutputSlot ( unsigned int index )

pure virtual

Get the output slot handle by slot index.

Implemented in Layer.

◆ GetParameters()

virtual const BaseDescriptor& GetParameters ( ) const

pure virtual

If the layer has a descriptor return it.

The base descriptor can then be cast to the correct descriptor class. If the layer has no associated descriptor a struct of type NullDescriptor will be returned. Note: NullDescriptors can be detected because they return true when the BaseDescriptor IsNull function is invoked.

Implemented in Layer, and LayerWithParameters< Parameters >.

◆ GetType()

virtual LayerType GetType ( ) const

pure virtual

Returns the armnn::LayerType of this layer.

Implemented in Layer.

Referenced by Connect(), SerializerStrategy::ExecuteStrategy(), StrategyBase< DefaultStrategy >::ExecuteStrategy(), SubgraphView::GetWorkingCopy(), and armnn_driver::IsConnectedToDequantize().

◆ InferOutputShapes()

virtual std::vector<TensorShape> InferOutputShapes ( const std::vector< TensorShape > & inputShapes ) const

pure virtual

Infer the shape of the output(s) based on the provided input shape(s)

Implemented in Layer, QLstmLayer, QuantizedLstmLayer, ConcatLayer, LstmLayer, UnidirectionalSequenceLstmLayer, SplitterLayer, DetectionPostProcessLayer, TransposeConvolution2dLayer, Convolution2dLayer, Convolution3dLayer, DepthwiseConvolution2dLayer, FullyConnectedLayer, ConstantLayer, ReshapeLayer, MeanLayer, PadLayer, StandInLayer, BatchToSpaceNdLayer, PermuteLayer, Pooling2dLayer, Pooling3dLayer, ResizeLayer, ShapeLayer, SliceLayer, StackLayer, MergeLayer, RankLayer, TransposeLayer, ComparisonLayer, DepthToSpaceLayer, LogicalBinaryLayer, PreluLayer, ReduceLayer, SpaceToBatchNdLayer, SpaceToDepthLayer, ArgMinMaxLayer, ElementwiseUnaryLayer, GatherLayer, StridedSliceLayer, BatchMatMulLayer, GatherNdLayer, and ElementwiseBaseLayer.

◆ SetBackendId()

virtual void SetBackendId ( const BackendId & id )

pure virtual

Set the backend of the IConnectableLayer.

By using SetBackendId() we guarantee that the input backend supports that layer (IsLayerSupported returns true for a specific backend). If there is no guarantee the input backend supports that layer use BackendSelectionHint().

Implemented in Layer.

Referenced by armnn_driver::ConvertPooling2d(), armnn_driver::ConvertReduce(), armnn_driver::ConvertToActivation(), armnn_driver::ConvertToLayerInputHandle(), and armnn_driver::ProcessActivation().

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

include/armnn/INetwork.hpp

Public Types

Public Member Functions

Protected Member Functions

Detailed Description

Member Typedef Documentation

◆ ConstantTensors

Constructor & Destructor Documentation

◆ ~IConnectableLayer()

Member Function Documentation

◆ BackendSelectionHint()

◆ ExecuteStrategy()

◆ GetConstantTensorsByRef()

◆ GetGuid()

◆ GetInputSlot() [1/2]

◆ GetInputSlot() [2/2]

◆ GetName()

◆ GetNumInputSlots()

◆ GetNumOutputSlots()

◆ GetOutputSlot() [1/2]

◆ GetOutputSlot() [2/2]

◆ GetParameters()

◆ GetType()

◆ InferOutputShapes()

◆ SetBackendId()