ArmNN
 22.02
LstmDescriptor Struct Reference

An LstmDescriptor for the LstmLayer. More...

#include <Descriptors.hpp>

Inheritance diagram for LstmDescriptor:
BaseDescriptor

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...
 

Detailed Description

An LstmDescriptor for the LstmLayer.

Definition at line 1078 of file Descriptors.hpp.

Constructor & Destructor Documentation

◆ LstmDescriptor()

LstmDescriptor ( )
inline

Definition at line 1080 of file Descriptors.hpp.

1081  : m_ActivationFunc(1) // 0: None, 1: Relu, 3: Relu6, 4: Tanh, 6: Sigmoid
1082  , m_ClippingThresCell(0.0)
1083  , m_ClippingThresProj(0.0)
1084  , m_CifgEnabled(true)
1085  , m_PeepholeEnabled(false)
1086  , m_ProjectionEnabled(false)
1087  , m_LayerNormEnabled(false)
1088  , m_TimeMajor(false)
1089  {}
bool m_ProjectionEnabled
Enable/disable the projection layer.
float m_ClippingThresProj
Clipping threshold value for the projection.
bool m_TimeMajor
Enable/disable time major.
bool m_PeepholeEnabled
Enable/disable peephole.
uint32_t m_ActivationFunc
The activation function to use.
float m_ClippingThresCell
Clipping threshold value for the cell state.
bool m_CifgEnabled
Enable/disable cifg (coupled input & forget gate).
bool m_LayerNormEnabled
Enable/disable layer normalization.

Member Function Documentation

◆ operator==()

bool operator== ( const LstmDescriptor rhs) const
inline

Definition at line 1091 of file Descriptors.hpp.

References LstmDescriptor::m_ActivationFunc, LstmDescriptor::m_CifgEnabled, LstmDescriptor::m_ClippingThresCell, LstmDescriptor::m_ClippingThresProj, LstmDescriptor::m_LayerNormEnabled, LstmDescriptor::m_PeepholeEnabled, and LstmDescriptor::m_TimeMajor.

1092  {
1093  return m_ActivationFunc == rhs.m_ActivationFunc &&
1094  m_ClippingThresCell == rhs.m_ClippingThresCell &&
1095  m_ClippingThresProj == rhs.m_ClippingThresProj &&
1096  m_CifgEnabled == rhs.m_CifgEnabled &&
1097  m_PeepholeEnabled == rhs.m_PeepholeEnabled &&
1098  m_LayerNormEnabled == rhs.m_LayerNormEnabled &&
1099  m_TimeMajor == rhs.m_TimeMajor;
1100  }
float m_ClippingThresProj
Clipping threshold value for the projection.
bool m_TimeMajor
Enable/disable time major.
bool m_PeepholeEnabled
Enable/disable peephole.
uint32_t m_ActivationFunc
The activation function to use.
float m_ClippingThresCell
Clipping threshold value for the cell state.
bool m_CifgEnabled
Enable/disable cifg (coupled input & forget gate).
bool m_LayerNormEnabled
Enable/disable layer normalization.

Member Data Documentation

◆ m_ActivationFunc

◆ m_CifgEnabled

bool m_CifgEnabled

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

Definition at line 1110 of file Descriptors.hpp.

Referenced by TestLstmLayerVisitor::CheckDescriptor(), LstmLayer::CreateWorkload(), UnidirectionalSequenceLstmLayer::CreateWorkload(), RefLstmWorkload::ExecuteAsync(), RefUnidirectionalSequenceLstmWorkload::ExecuteAsync(), LstmLayer::ExecuteStrategy(), UnidirectionalSequenceLstmLayer::ExecuteStrategy(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), LstmLayer::InferOutputShapes(), RefLayerSupport::IsLstmSupported(), RefLayerSupport::IsUnidirectionalSequenceLstmSupported(), armnn::LstmImpl(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), TEST_SUITE(), armnn::TEST_SUITE(), UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjectionTest(), UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest(), UnidirectionalSequenceLstmLayerInt8Test(), UnidirectionalSequenceLstmLayerInt8TimeMajorTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest(), LstmLayer::ValidateTensorShapesFromInputs(), and UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs().

◆ m_ClippingThresCell

◆ m_ClippingThresProj

◆ m_LayerNormEnabled

◆ m_PeepholeEnabled

◆ m_ProjectionEnabled

◆ m_TimeMajor


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