This layer represents a split operation. More...

#include <SplitterLayer.hpp>

Inheritance diagram for SplitterLayer:

Public Member Functions
virtual std::unique_ptr< IWorkload >	CreateWorkload (const IWorkloadFactory &factory) const override

virtual void	CreateTensorHandles (const TensorHandleFactoryRegistry &registry, const IWorkloadFactory &factory, const bool IsMemoryManaged=true) override

SplitterLayer *	Clone (Graph &graph) const override

void	ValidateTensorShapesFromInputs () override

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

void	Accept (ILayerVisitor &visitor) const override

Public Member Functions inherited from LayerWithParameters< ViewsDescriptor >
const ViewsDescriptor &	GetParameters () const

void	SerializeLayerParameters (ParameterStringifyFunction &fn) const override

Public Member Functions inherited from Layer
	Layer (unsigned int numInputSlots, unsigned int numOutputSlots, LayerType type, const char *name)

	Layer (unsigned int numInputSlots, unsigned int numOutputSlots, LayerType type, DataLayout layout, const char *name)

const std::string &	GetNameStr () const

const OutputHandler &	GetOutputHandler (unsigned int i=0) const

OutputHandler &	GetOutputHandler (unsigned int i=0)

const std::vector< InputSlot > &	GetInputSlots () const

const std::vector< OutputSlot > &	GetOutputSlots () const

std::vector< InputSlot >::iterator	BeginInputSlots ()

std::vector< InputSlot >::iterator	EndInputSlots ()

std::vector< OutputSlot >::iterator	BeginOutputSlots ()

std::vector< OutputSlot >::iterator	EndOutputSlots ()

bool	IsOutputUnconnected ()

void	ResetPriority () const

LayerPriority	GetPriority () const

LayerType	GetType () const

DataType	GetDataType () const

const BackendId &	GetBackendId () const

void	SetBackendId (const BackendId &id)

void	VerifyLayerConnections (unsigned int expectedConnections, const CheckLocation &location) const

virtual void	ReleaseConstantData ()

template<typename Op >
void	OperateOnConstantTensors (Op op)

const char *	GetName () const override

unsigned int	GetNumInputSlots () const override

unsigned int	GetNumOutputSlots () const override

const InputSlot &	GetInputSlot (unsigned int index) const override

InputSlot &	GetInputSlot (unsigned int index) override

const OutputSlot &	GetOutputSlot (unsigned int index=0) const override

OutputSlot &	GetOutputSlot (unsigned int index=0) override

void	SetGuid (LayerGuid guid)

LayerGuid	GetGuid () const final

void	AddRelatedLayerName (const std::string layerName)

const std::list< std::string > &	GetRelatedLayerNames ()

virtual void	Reparent (Graph &dest, std::list< Layer *>::const_iterator iterator)=0

Protected Member Functions
	SplitterLayer (const ViewsDescriptor &param, const char *name)

	~SplitterLayer ()=default
	Default destructor. More...

Protected Member Functions inherited from LayerWithParameters< ViewsDescriptor >
	LayerWithParameters (unsigned int numInputSlots, unsigned int numOutputSlots, LayerType type, const ViewsDescriptor &param, const char *name)

	~LayerWithParameters ()=default

WorkloadInfo	PrepInfoAndDesc (QueueDescriptor &descriptor) const
	Helper function to reduce duplication in *LayerCreateWorkload. More...

Protected Member Functions inherited from Layer
virtual	~Layer ()=default

template<typename QueueDescriptor >
void	CollectQueueDescriptorInputs (QueueDescriptor &descriptor, WorkloadInfo &info) const

template<typename QueueDescriptor >
void	CollectQueueDescriptorOutputs (QueueDescriptor &descriptor, WorkloadInfo &info) const

template<typename QueueDescriptor >
WorkloadInfo	PrepInfoAndDesc (QueueDescriptor &descriptor) const
	Helper function to reduce duplication in *LayerCreateWorkload. More...

template<typename LayerType , typename ... Params>
LayerType *	CloneBase (Graph &graph, Params &&... params) const

virtual ConstantTensors	GetConstantTensorsByRef ()

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

Additional Inherited Members
Public Types inherited from LayerWithParameters< ViewsDescriptor >
using	DescriptorType = ViewsDescriptor

Protected Types inherited from Layer
using	ConstantTensors = std::vector< std::reference_wrapper< std::unique_ptr< ScopedCpuTensorHandle > >>

Protected Attributes inherited from LayerWithParameters< ViewsDescriptor >
ViewsDescriptor	m_Param
	The parameters for the layer (not including tensor-valued weights etc.). More...

Protected Attributes inherited from Layer
std::vector< OutputHandler >	m_OutputHandlers

Detailed Description

This layer represents a split operation.

Definition at line 13 of file SplitterLayer.hpp.

Constructor & Destructor Documentation

◆ SplitterLayer()

SplitterLayer	(	const ViewsDescriptor &	param,
		const char *	name
	)

protected

Constructor to create a SplitterLayer.

Parameters

[in]	param	ViewsDescriptor to configure the splitter operation.
[in]	name	Optional name for the layer.

Definition at line 16 of file SplitterLayer.cpp.

     : LayerWithParameters(1, param.GetNumViews(), LayerType::Splitter, param, name)
 {
 }

◆ ~SplitterLayer()

~SplitterLayer ( )

protecteddefault

Default destructor.

Member Function Documentation

◆ Accept()

void Accept ( ILayerVisitor & visitor ) const

overridevirtual

Implements IConnectableLayer.

Definition at line 164 of file SplitterLayer.cpp.

References Layer::GetName(), LayerWithParameters< ViewsDescriptor >::GetParameters(), and ILayerVisitor::VisitSplitterLayer().

 {
     visitor.VisitSplitterLayer(this, GetParameters(), GetName());
 }

◆ Clone()

SplitterLayer * Clone ( Graph & graph ) const

overridevirtual

Creates a dynamically-allocated copy of this layer.

Parameters

[in] graph The graph into which this layer is being cloned.

Implements Layer.

Definition at line 123 of file SplitterLayer.cpp.

References Layer::GetName(), and LayerWithParameters< ViewsDescriptor >::m_Param.

 {
     return CloneBase<SplitterLayer>(graph, m_Param, GetName());
 }

◆ CreateTensorHandles()

void CreateTensorHandles	(	const TensorHandleFactoryRegistry &	registry,
		const IWorkloadFactory &	factory,
		const bool	IsMemoryManaged = `true`
	)

overridevirtual

Set the outputs to be appropriate sub tensors of the input if sub tensors are supported otherwise creates tensor handlers.

Parameters

[in]	registry	Contains all the registered tensor handle factories available for use.
[in]	factory	The workload factory which will create the workload.
[in]	IsMemoryManaged	Determine whether or not to assign a memory manager during creation

Reimplemented from Layer.

Definition at line 103 of file SplitterLayer.cpp.

References TensorHandleFactoryRegistry::GetFactory(), Layer::GetOutputSlot(), OutputSlot::GetTensorHandleFactoryId(), and ITensorHandleFactory::LegacyFactoryId.

 {
     boost::ignore_unused(IsMemoryManaged);
     OutputSlot& slot = GetOutputSlot(0);
     ITensorHandleFactory::FactoryId factoryId = slot.GetTensorHandleFactoryId();
 
     if (factoryId == ITensorHandleFactory::LegacyFactoryId)
     {
         CreateTensors(workloadFactory);
     }
     else
     {
         ITensorHandleFactory* handleFactory = registry.GetFactory(factoryId);
         BOOST_ASSERT(handleFactory);
         CreateTensors(*handleFactory);
     }
 }

◆ CreateWorkload()

std::unique_ptr< IWorkload > CreateWorkload ( const IWorkloadFactory & factory ) const

overridevirtual

Makes a workload for the Splitter type.

Parameters

[in]	graph	The graph where this layer can be found.
[in]	factory	The workload factory which will create the workload.

Returns: A pointer to the created workload, or nullptr if not created.

Implements Layer.

Definition at line 21 of file SplitterLayer.cpp.

References IWorkloadFactory::CreateSplitter(), OutputHandler::GetData(), Layer::GetInputSlots(), ViewsDescriptor::GetNumDimensions(), ViewsDescriptor::GetNumViews(), Layer::GetOutputSlot(), TensorInfo::GetShape(), OutputSlot::GetTensorHandleFactoryId(), OutputHandler::GetTensorInfo(), ViewsDescriptor::GetViewOrigin(), armnn::info, TensorInfo::IsTypeSpaceMatch(), Layer::m_OutputHandlers, LayerWithParameters< ViewsDescriptor >::m_Param, SplitterQueueDescriptor::m_ViewOrigins, and LayerWithParameters< ViewsDescriptor >::PrepInfoAndDesc().

 {
     SplitterQueueDescriptor descriptor;
 
     // Copies the window origins to the descriptor.
     for (unsigned int i = 0; i < m_Param.GetNumViews(); ++i)
     {
         descriptor.m_ViewOrigins.emplace_back(
             std::vector<unsigned int>(m_Param.GetViewOrigin(i), m_Param.GetViewOrigin(i) + m_Param.GetNumDimensions()));
     }
 
     return factory.CreateSplitter(descriptor, PrepInfoAndDesc(descriptor));
 }

◆ InferOutputShapes()

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

overridevirtual

By default returns inputShapes if the number of inputs are equal to number of outputs, otherwise infers the output shapes from given input shapes and layer properties.

Parameters

[in] inputShapes The input shapes layer has.

Returns: A vector to the inferred output shape.

Reimplemented from Layer.

Definition at line 128 of file SplitterLayer.cpp.

References ViewsDescriptor::GetNumDimensions(), ViewsDescriptor::GetNumViews(), ViewsDescriptor::GetViewSizes(), and LayerWithParameters< ViewsDescriptor >::m_Param.

Referenced by SplitterLayer::ValidateTensorShapesFromInputs().

 {
     boost::ignore_unused(inputShapes);
     BOOST_ASSERT(inputShapes.size() ==  m_Param.GetNumViews());
     std::vector<TensorShape> outShapes;
     //Output shapes must match View shapes.
     for (unsigned int viewIdx = 0; viewIdx < m_Param.GetNumViews(); viewIdx++)
     {
         const uint32_t* sizes = m_Param.GetViewSizes(viewIdx);
         outShapes.push_back(TensorShape(m_Param.GetNumDimensions(), sizes));
     }
     return outShapes;
 }

◆ ValidateTensorShapesFromInputs()

void ValidateTensorShapesFromInputs ( )

overridevirtual

Check if the input tensor shape(s) will lead to a valid configuration of SplitterLayer.

Implements Layer.

Definition at line 142 of file SplitterLayer.cpp.

References ViewsDescriptor::GetNumDimensions(), ViewsDescriptor::GetNumViews(), Layer::GetOutputSlot(), TensorInfo::GetShape(), OutputSlot::GetTensorInfo(), ViewsDescriptor::GetViewSizes(), SplitterLayer::InferOutputShapes(), and LayerWithParameters< ViewsDescriptor >::m_Param.

 {
     std::vector<TensorShape> views;
     for (unsigned int viewIdx = 0; viewIdx < m_Param.GetNumViews(); viewIdx++)
     {
         const uint32_t* sizes = m_Param.GetViewSizes(viewIdx);
         views.push_back(TensorShape(m_Param.GetNumDimensions(), sizes));
     }
 
     auto inferredShapes = InferOutputShapes(views);
 
     BOOST_ASSERT(inferredShapes.size() == m_Param.GetNumViews());
 
     for (unsigned int viewIdx = 0; viewIdx < m_Param.GetNumViews(); viewIdx++)
     {
         ConditionalThrowIfNotEqual<LayerValidationException>(
             "SplitterLayer: View sizes must match output tensor shapes.",
             GetOutputSlot(viewIdx).GetTensorInfo().GetShape(),
             inferredShapes[viewIdx]);
     }
 }

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

src/armnn/layers/SplitterLayer.hpp
src/armnn/layers/SplitterLayer.cpp

Public Member Functions

Protected Member Functions

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ SplitterLayer()

◆ ~SplitterLayer()

Member Function Documentation

◆ Accept()

◆ Clone()

◆ CreateTensorHandles()

◆ CreateWorkload()

◆ InferOutputShapes()

◆ ValidateTensorShapesFromInputs()