LayerWithParameters< Parameters > Class Template Reference

#include <LayerWithParameters.hpp>

Inheritance diagram for LayerWithParameters< Parameters >:

Public Types
using	DescriptorType = Parameters

Public Member Functions
const Parameters &	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 std::unique_ptr< IWorkload >	CreateWorkload (const IWorkloadFactory &factory) const =0
virtual void	CreateTensorHandles (const TensorHandleFactoryRegistry &registry, const IWorkloadFactory &factory, const bool IsMemoryManaged=true)
virtual Layer *	Clone (Graph &graph) const =0
void	VerifyLayerConnections (unsigned int expectedConnections, const CheckLocation &location) const
virtual void	ValidateTensorShapesFromInputs ()=0
std::vector< TensorShape >	InferOutputShapes (const std::vector< TensorShape > &inputShapes) const override
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
Public Member Functions inherited from IConnectableLayer
virtual void	Accept (ILayerVisitor &visitor) const =0

Protected Member Functions
	LayerWithParameters (unsigned int numInputSlots, unsigned int numOutputSlots, LayerType type, const Parameters &param, const char *name)
	~LayerWithParameters ()=default
template<typename QueueDescriptor >
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...

Protected Attributes
Parameters	m_Param
	The parameters for the layer (not including tensor-valued weights etc.). More...
Protected Attributes inherited from Layer
std::vector< OutputHandler >	m_OutputHandlers

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

Detailed Description

template<typename Parameters>
class armnn::LayerWithParameters< Parameters >

Definition at line 13 of file LayerWithParameters.hpp.

Member Typedef Documentation

DescriptorType

using DescriptorType = Parameters

Definition at line 16 of file LayerWithParameters.hpp.

Constructor & Destructor Documentation

LayerWithParameters()

LayerWithParameters ( unsigned int  numInputSlots, unsigned int  numOutputSlots, LayerType  type, const Parameters &  param, const char *  name  )

inlineprotected

Definition at line 29 of file LayerWithParameters.hpp.

        : Layer(numInputSlots, numOutputSlots, type, name)
        , m_Param(param)
    {
    }

~LayerWithParameters()

~LayerWithParameters ( )

protecteddefault

Referenced by LayerWithParameters< ResizeDescriptor >::LayerWithParameters().

Member Function Documentation

GetParameters()

const Parameters& GetParameters ( ) const

inline

Definition at line 18 of file LayerWithParameters.hpp.

Referenced by BOOST_AUTO_TEST_CASE().

{ return m_Param; }

PrepInfoAndDesc()

WorkloadInfo PrepInfoAndDesc ( QueueDescriptor &  descriptor ) const

inlineprotected

Helper function to reduce duplication in *LayerCreateWorkload.

Definition at line 43 of file LayerWithParameters.hpp.

    {
        descriptor.m_Parameters = m_Param;
        return Layer::PrepInfoAndDesc(descriptor);
    }

SerializeLayerParameters()

void SerializeLayerParameters ( ParameterStringifyFunction &  fn ) const

inlineoverridevirtual

Helper to serialize the layer parameters to string (currently used in DotSerializer and company).

Reimplemented from Layer.

Definition at line 22 of file LayerWithParameters.hpp.

Referenced by DepthwiseConvolution2dLayer::SerializeLayerParameters(), and Convolution2dLayer::SerializeLayerParameters().

    {
        StringifyLayerParameters<Parameters>::Serialize(fn, m_Param);
        Layer::SerializeLayerParameters(fn);
    }

Member Data Documentation

m_Param

Parameters m_Param

protected

The parameters for the layer (not including tensor-valued weights etc.).

Definition at line 50 of file LayerWithParameters.hpp.

Referenced by MeanLayer::Clone(), PadLayer::Clone(), ConcatLayer::CreateWorkload(), LayerWithParameters< ResizeDescriptor >::GetParameters(), LayerWithParameters< ResizeDescriptor >::PrepInfoAndDesc(), and LayerWithParameters< ResizeDescriptor >::SerializeLayerParameters().

---

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

• src/armnn/layers/LayerWithParameters.hpp