LstmDescriptor Struct Reference

An LstmDescriptor for the LstmLayer. More...

#include <Descriptors.hpp>

Inheritance diagram for LstmDescriptor:

Collaboration diagram for LstmDescriptor:

Public Member Functions
	LstmDescriptor ()
bool	operator== (const LstmDescriptor &rhs) const
Public Member Functions inherited from BaseDescriptor
virtual bool	IsNull () const
virtual	~BaseDescriptor ()=default

Public Attributes
uint32_t	m_ActivationFunc
	The activation function to use. More...
float	m_ClippingThresCell
	Clipping threshold value for the cell state. More...
float	m_ClippingThresProj
	Clipping threshold value for the projection. More...
bool	m_CifgEnabled
	Enable/disable cifg (coupled input & forget gate). More...
bool	m_PeepholeEnabled
	Enable/disable peephole. More...
bool	m_ProjectionEnabled
	Enable/disable the projection layer. More...
bool	m_LayerNormEnabled
	Enable/disable layer normalization. More...
bool	m_TimeMajor
	Enable/disable time major. More...
float	m_InputIntermediateScale
	Input intermediate quantization scale. More...
float	m_ForgetIntermediateScale
	Forget intermediate quantization scale. More...
float	m_CellIntermediateScale
	Cell intermediate quantization scale. More...
float	m_OutputIntermediateScale
	Output intermediate quantization scale. More...
int32_t	m_HiddenStateZeroPoint
	Hidden State zero point. More...
float	m_HiddenStateScale
	Hidden State quantization scale. More...

Detailed Description

An LstmDescriptor for the LstmLayer.

Definition at line 1102 of file Descriptors.hpp.

Constructor & Destructor Documentation

LstmDescriptor()

LstmDescriptor ( )

inline

Definition at line 1104 of file Descriptors.hpp.

        : m_ActivationFunc(1) // 0: None, 1: Relu, 3: Relu6, 4: Tanh, 6: Sigmoid
        , m_ClippingThresCell(0.0)
        , m_ClippingThresProj(0.0)
        , m_CifgEnabled(true)
        , m_PeepholeEnabled(false)
        , m_ProjectionEnabled(false)
        , m_LayerNormEnabled(false)
        , m_TimeMajor(false)
        , m_InputIntermediateScale(0.0)
        , m_ForgetIntermediateScale(0.0)
        , m_CellIntermediateScale(0.0)
        , m_OutputIntermediateScale(0.0)
        , m_HiddenStateZeroPoint(0)
        , m_HiddenStateScale(0.0)
    {}

Member Function Documentation

operator==()

bool operator== ( const LstmDescriptor &  rhs ) const

inline

Definition at line 1121 of file Descriptors.hpp.

    {
        return m_ActivationFunc          == rhs.m_ActivationFunc &&
               m_ClippingThresCell       == rhs.m_ClippingThresCell &&
               m_ClippingThresProj       == rhs.m_ClippingThresProj &&
               m_CifgEnabled             == rhs.m_CifgEnabled &&
               m_PeepholeEnabled         == rhs.m_PeepholeEnabled &&
               m_LayerNormEnabled        == rhs.m_LayerNormEnabled &&
               m_TimeMajor               == rhs.m_TimeMajor &&
               m_InputIntermediateScale  == rhs.m_InputIntermediateScale &&
               m_ForgetIntermediateScale == rhs.m_ForgetIntermediateScale &&
               m_CellIntermediateScale   == rhs.m_CellIntermediateScale &&
               m_OutputIntermediateScale == rhs.m_OutputIntermediateScale &&
               m_HiddenStateZeroPoint    == rhs.m_HiddenStateZeroPoint &&
               m_HiddenStateScale        == rhs.m_HiddenStateScale;
    }

References LstmDescriptor::m_ActivationFunc, LstmDescriptor::m_CellIntermediateScale, LstmDescriptor::m_CifgEnabled, LstmDescriptor::m_ClippingThresCell, LstmDescriptor::m_ClippingThresProj, LstmDescriptor::m_ForgetIntermediateScale, LstmDescriptor::m_HiddenStateScale, LstmDescriptor::m_HiddenStateZeroPoint, LstmDescriptor::m_InputIntermediateScale, LstmDescriptor::m_LayerNormEnabled, LstmDescriptor::m_OutputIntermediateScale, LstmDescriptor::m_PeepholeEnabled, and LstmDescriptor::m_TimeMajor.

Member Data Documentation

m_ActivationFunc

uint32_t m_ActivationFunc

The activation function to use.

0: None, 1: Relu, 3: Relu6, 4: Tanh, 6: Sigmoid.

Definition at line 1140 of file Descriptors.hpp.

Referenced by IDeserializer::DeserializerImpl::GetLstmDescriptor(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), armnn::LstmImpl(), LstmDescriptor::operator==(), and StringifyLayerParameters< LstmDescriptor >::Serialize().

m_CellIntermediateScale

float m_CellIntermediateScale

Cell intermediate quantization scale.

Definition at line 1160 of file Descriptors.hpp.

Referenced by LstmDescriptor::operator==().

m_CifgEnabled

bool m_CifgEnabled

Enable/disable cifg (coupled input & forget gate).

Definition at line 1146 of file Descriptors.hpp.

Referenced by LstmLayer::Clone(), UnidirectionalSequenceLstmLayer::Clone(), LstmLayer::CreateWorkload(), UnidirectionalSequenceLstmLayer::CreateWorkload(), LstmLayer::ExecuteStrategy(), UnidirectionalSequenceLstmLayer::ExecuteStrategy(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), LstmLayer::InferOutputShapes(), RefLayerSupport::IsLstmSupported(), RefLayerSupport::IsUnidirectionalSequenceLstmSupported(), armnn::LstmImpl(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), LstmQueueDescriptor::Validate(), UnidirectionalSequenceLstmQueueDescriptor::Validate(), UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs(), and LstmLayer::ValidateTensorShapesFromInputs().

m_ClippingThresCell

float m_ClippingThresCell

Clipping threshold value for the cell state.

Definition at line 1142 of file Descriptors.hpp.

Referenced by IDeserializer::DeserializerImpl::GetLstmDescriptor(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), armnn::LstmImpl(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), LstmQueueDescriptor::Validate(), and UnidirectionalSequenceLstmQueueDescriptor::Validate().

m_ClippingThresProj

float m_ClippingThresProj

Clipping threshold value for the projection.

Definition at line 1144 of file Descriptors.hpp.

Referenced by IDeserializer::DeserializerImpl::GetLstmDescriptor(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), armnn::LstmImpl(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), LstmQueueDescriptor::Validate(), and UnidirectionalSequenceLstmQueueDescriptor::Validate().

m_ForgetIntermediateScale

float m_ForgetIntermediateScale

Forget intermediate quantization scale.

Definition at line 1158 of file Descriptors.hpp.

Referenced by LstmDescriptor::operator==().

m_HiddenStateScale

float m_HiddenStateScale

Hidden State quantization scale.

Definition at line 1166 of file Descriptors.hpp.

Referenced by LstmDescriptor::operator==().

m_HiddenStateZeroPoint

int32_t m_HiddenStateZeroPoint

Hidden State zero point.

Definition at line 1164 of file Descriptors.hpp.

Referenced by LstmDescriptor::operator==().

m_InputIntermediateScale

float m_InputIntermediateScale

Input intermediate quantization scale.

Definition at line 1156 of file Descriptors.hpp.

Referenced by LstmDescriptor::operator==().

m_LayerNormEnabled

bool m_LayerNormEnabled

Enable/disable layer normalization.

Definition at line 1152 of file Descriptors.hpp.

Referenced by LstmLayer::Clone(), UnidirectionalSequenceLstmLayer::Clone(), UnidirectionalSequenceLstmLayer::CreateWorkload(), LstmLayer::CreateWorkload(), LstmLayer::ExecuteStrategy(), UnidirectionalSequenceLstmLayer::ExecuteStrategy(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), RefLayerSupport::IsLstmSupported(), RefLayerSupport::IsUnidirectionalSequenceLstmSupported(), armnn::LstmImpl(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), LstmQueueDescriptor::Validate(), UnidirectionalSequenceLstmQueueDescriptor::Validate(), and UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs().

m_OutputIntermediateScale

float m_OutputIntermediateScale

Output intermediate quantization scale.

Definition at line 1162 of file Descriptors.hpp.

Referenced by LstmDescriptor::operator==().

m_PeepholeEnabled

bool m_PeepholeEnabled

Enable/disable peephole.

Definition at line 1148 of file Descriptors.hpp.

Referenced by LstmLayer::Clone(), UnidirectionalSequenceLstmLayer::Clone(), UnidirectionalSequenceLstmLayer::CreateWorkload(), LstmLayer::CreateWorkload(), LstmLayer::ExecuteStrategy(), UnidirectionalSequenceLstmLayer::ExecuteStrategy(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), RefLayerSupport::IsLstmSupported(), RefLayerSupport::IsUnidirectionalSequenceLstmSupported(), armnn::LstmImpl(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), LstmQueueDescriptor::Validate(), UnidirectionalSequenceLstmQueueDescriptor::Validate(), UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs(), and LstmLayer::ValidateTensorShapesFromInputs().

m_ProjectionEnabled

bool m_ProjectionEnabled

Enable/disable the projection layer.

Definition at line 1150 of file Descriptors.hpp.

Referenced by LstmLayer::Clone(), UnidirectionalSequenceLstmLayer::Clone(), LstmLayer::CreateWorkload(), UnidirectionalSequenceLstmLayer::CreateWorkload(), LstmLayer::ExecuteStrategy(), UnidirectionalSequenceLstmLayer::ExecuteStrategy(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), RefLayerSupport::IsLstmSupported(), RefLayerSupport::IsUnidirectionalSequenceLstmSupported(), armnn::LstmImpl(), StringifyLayerParameters< LstmDescriptor >::Serialize(), LstmQueueDescriptor::Validate(), UnidirectionalSequenceLstmQueueDescriptor::Validate(), LstmLayer::ValidateTensorShapesFromInputs(), and UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs().

m_TimeMajor

bool m_TimeMajor

Enable/disable time major.

Definition at line 1154 of file Descriptors.hpp.

Referenced by armnn::ClUnidirectionalSequenceLstmFloatWorkloadValidate(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), UnidirectionalSequenceLstmLayer::InferOutputShapes(), armnn::NeonUnidirectionalSequenceLstmFloatWorkloadValidate(), armnn::NeonUnidirectionalSequenceLstmWorkloadValidate(), LstmDescriptor::operator==(), and UnidirectionalSequenceLstmQueueDescriptor::Validate().

---

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

• include/armnn/Descriptors.hpp