#include <PreluLayer.hpp>

Inheritance diagram for PreluLayer:

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

PreluLayer *	Clone (Graph &graph) const override

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

void	ValidateTensorShapesFromInputs () override

void	Accept (ILayerVisitor &visitor) 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	SerializeLayerParameters (ParameterStringifyFunction &fn) 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
	PreluLayer (const char *name)

	~PreluLayer ()=default
	Default destructor. 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
Protected Types inherited from Layer
using	ConstantTensors = std::vector< std::reference_wrapper< std::unique_ptr< ScopedCpuTensorHandle > >>

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

Detailed Description

Definition at line 14 of file PreluLayer.hpp.

Constructor & Destructor Documentation

◆ PreluLayer()

PreluLayer ( const char * name )

protected

Constructor to create a PreluLayer.

Parameters

[in] name Optional name for the layer.

Definition at line 17 of file PreluLayer.cpp.

References armnn::Prelu.

18 : Layer(2, 1, LayerType::Prelu, name)

19 {}

armnn::Layer::Layer

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

Definition: Layer.cpp:212

armnn::LayerType::Prelu

◆ ~PreluLayer()

~PreluLayer ( )

protecteddefault

Default destructor.

Member Function Documentation

◆ Accept()

void Accept ( ILayerVisitor & visitor ) const

overridevirtual

Implements IConnectableLayer.

Definition at line 115 of file PreluLayer.cpp.

References Layer::GetName(), and ILayerVisitor::VisitPreluLayer().

 {
     visitor.VisitPreluLayer(this, GetName());
 }

◆ Clone()

PreluLayer * 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 28 of file PreluLayer.cpp.

References Layer::GetName().

 {
     auto layer = CloneBase<PreluLayer>(graph, GetName());
 
     return std::move(layer);
 }

◆ CreateWorkload()

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

overridevirtual

Makes a workload for the PReLU 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 PreluLayer.cpp.

References IWorkloadFactory::CreatePrelu(), and Layer::PrepInfoAndDesc().

 {
     PreluQueueDescriptor descriptor;
 
     return factory.CreatePrelu(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 35 of file PreluLayer.cpp.

References TensorShape::GetNumDimensions().

Referenced by PreluInferOutputShapeImpl(), and PreluLayer::ValidateTensorShapesFromInputs().

 {
     BOOST_ASSERT(inputShapes.size() == 2);
 
     const TensorShape& inputShape = inputShapes[0];
     const TensorShape& alphaShape = inputShapes[1];
 
     const unsigned int inputShapeDimensions = inputShape.GetNumDimensions();
     const unsigned int alphaShapeDimensions = alphaShape.GetNumDimensions();
 
     BOOST_ASSERT(inputShapeDimensions > 0);
     BOOST_ASSERT(alphaShapeDimensions > 0);
 
     // The size of the output is the maximum size along each dimension of the input operands,
     // it starts with the trailing dimensions, and works its way forward
 
     unsigned int outputDimensions = std::max(inputShapeDimensions, alphaShapeDimensions);
 
     TensorShape outputShape(outputDimensions);
 
     int inputShapeIndex = boost::numeric_cast<int>(inputShapeDimensions) - 1;
     int alphaShapeIndex = boost::numeric_cast<int>(alphaShapeDimensions) - 1;
     unsigned int outputShapeIndex = outputDimensions - 1;
 
     // Loop backwards through the common part of the shapes
     while (inputShapeIndex >= 0 && alphaShapeIndex >= 0)
     {
         unsigned int inputDimension = inputShape[boost::numeric_cast<unsigned int>(inputShapeIndex)];
         unsigned int alphaDimension = alphaShape[boost::numeric_cast<unsigned int>(alphaShapeIndex)];
 
         // Check that the inputs are broadcast compatible
         BOOST_ASSERT_MSG(inputDimension == alphaDimension || inputDimension == 1 || alphaDimension == 1,
                          "PreluLayer: Dimensions should either match or one should be of size 1");
 
         outputShape[outputShapeIndex] = std::max(inputDimension, alphaDimension);
 
         inputShapeIndex--;
         alphaShapeIndex--;
         outputShapeIndex--;
     }
 
     // Loop backwards through the remaing part of the input shape (if any)
     while (inputShapeIndex >= 0)
     {
         outputShape[outputShapeIndex] = inputShape[boost::numeric_cast<unsigned int>(inputShapeIndex)];
 
         inputShapeIndex--;
         outputShapeIndex--;
     }
 
     // Loop backwards through the remaing part of the alpha shape (if any)
     while (alphaShapeIndex >= 0)
     {
         outputShape[outputShapeIndex] = alphaShape[boost::numeric_cast<unsigned int>(alphaShapeIndex)];
 
         alphaShapeIndex--;
         outputShapeIndex--;
     }
 
     return { outputShape };
 }

◆ ValidateTensorShapesFromInputs()

void ValidateTensorShapesFromInputs ( )

overridevirtual

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

Implements Layer.

Definition at line 97 of file PreluLayer.cpp.

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

 {
     VerifyLayerConnections(2, CHECK_LOCATION());
 
     std::vector<TensorShape> inferredShapes = InferOutputShapes(
     {
         GetInputSlot(0).GetConnection()->GetTensorInfo().GetShape(),
         GetInputSlot(1).GetConnection()->GetTensorInfo().GetShape()
     });
 
     BOOST_ASSERT(inferredShapes.size() == 1);
 
     ConditionalThrowIfNotEqual<LayerValidationException>(
         "PreluLayer: 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/PreluLayer.hpp
src/armnn/layers/PreluLayer.cpp

Public Member Functions

Protected Member Functions

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ PreluLayer()

◆ ~PreluLayer()

Member Function Documentation

◆ Accept()

◆ Clone()

◆ CreateWorkload()

◆ InferOutputShapes()

◆ ValidateTensorShapesFromInputs()