SliceLayer Class Reference

#include <SliceLayer.hpp>

Inheritance diagram for SliceLayer:

Public Member Functions
virtual std::unique_ptr< IWorkload >	CreateWorkload (const IWorkloadFactory &factory) const override
SliceLayer *	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< SliceDescriptor >
const SliceDescriptor &	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)
virtual void	CreateTensorHandles (const TensorHandleFactoryRegistry &registry, const IWorkloadFactory &factory, const bool IsMemoryManaged=true)
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
	SliceLayer (const SliceDescriptor &param, const char *name)
	~SliceLayer ()=default
	Default destructor. More...
Protected Member Functions inherited from LayerWithParameters< SliceDescriptor >
	LayerWithParameters (unsigned int numInputSlots, unsigned int numOutputSlots, LayerType type, const SliceDescriptor &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< SliceDescriptor >
using	DescriptorType = SliceDescriptor
Protected Types inherited from Layer
using	ConstantTensors = std::vector< std::reference_wrapper< std::unique_ptr< ScopedCpuTensorHandle > >>
Protected Attributes inherited from LayerWithParameters< SliceDescriptor >
SliceDescriptor	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

Definition at line 13 of file SliceLayer.hpp.

Constructor & Destructor Documentation

SliceLayer()

SliceLayer ( const SliceDescriptor &  param, const char *  name  )

protected

Constructor to create a SliceLayer.

Parameters

[in]	param	SliceDescriptor to configure the resize operation.
[in]	name	Optional name for the layer.

Definition at line 21 of file SliceLayer.cpp.

    : LayerWithParameters(1, 1, LayerType::Slice, param, name)
{
}

~SliceLayer()

~SliceLayer ( )

protecteddefault

Default destructor.

Member Function Documentation

Accept()

void Accept ( ILayerVisitor &  visitor ) const

overridevirtual

Implements IConnectableLayer.

Definition at line 61 of file SliceLayer.cpp.

References Layer::GetName(), LayerWithParameters< SliceDescriptor >::GetParameters(), and ILayerVisitor::VisitSliceLayer().

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

Clone()

SliceLayer * 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 32 of file SliceLayer.cpp.

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

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

CreateWorkload()

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

overridevirtual

Makes a workload for the Slice 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 26 of file SliceLayer.cpp.

References IWorkloadFactory::CreateSlice(), and LayerWithParameters< SliceDescriptor >::PrepInfoAndDesc().

{
    SliceQueueDescriptor descriptor;
    return factory.CreateSlice(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 51 of file SliceLayer.cpp.

References LayerWithParameters< SliceDescriptor >::m_Param, and SliceDescriptor::m_Size.

Referenced by SliceLayer::ValidateTensorShapesFromInputs().

{
    boost::ignore_unused(inputShapes);
    BOOST_ASSERT(inputShapes.size() == 1);

    TensorShape outputShape(boost::numeric_cast<unsigned int>(m_Param.m_Size.size()), m_Param.m_Size.data());

    return std::vector<TensorShape>({ outputShape });
}

ValidateTensorShapesFromInputs()

void ValidateTensorShapesFromInputs ( )

overridevirtual

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

Implements Layer.

Definition at line 37 of file SliceLayer.cpp.

References CHECK_LOCATION, InputSlot::GetConnection(), Layer::GetInputSlot(), Layer::GetOutputSlot(), TensorInfo::GetShape(), IOutputSlot::GetTensorInfo(), OutputSlot::GetTensorInfo(), SliceLayer::InferOutputShapes(), and Layer::VerifyLayerConnections().

{
    VerifyLayerConnections(1, CHECK_LOCATION());

    auto inferredShapes = InferOutputShapes({ GetInputSlot(0).GetConnection()->GetTensorInfo().GetShape() });

    BOOST_ASSERT(inferredShapes.size() == 1);

    ConditionalThrowIfNotEqual<LayerValidationException>(
            "SliceLayer: TensorShape set on OutputSlot[0] does not match the inferred shape.",
            GetOutputSlot(0).GetTensorInfo().GetShape(),
            inferredShapes[0]);
}

---

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

• src/armnn/layers/SliceLayer.hpp
• src/armnn/layers/SliceLayer.cpp