LstmDescriptor Struct Reference

An LstmDescriptor for the LstmLayer. More...

#include <Descriptors.hpp>

Public Member Functions
	LstmDescriptor ()
bool	operator== (const LstmDescriptor &rhs) const

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

Detailed Description

An LstmDescriptor for the LstmLayer.

Definition at line 837 of file Descriptors.hpp.

Constructor & Destructor Documentation

LstmDescriptor()

LstmDescriptor ( )

inline

Definition at line 839 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)
    {}

Member Function Documentation

operator==()

bool operator== ( const LstmDescriptor &  rhs ) const

inline

Definition at line 849 of file Descriptors.hpp.

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

    {
        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;
    }

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 861 of file Descriptors.hpp.

Referenced by armnn::BOOST_AUTO_TEST_CASE(), BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), RefLstmWorkload::Execute(), Deserializer::GetLstmDescriptor(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), and SerializerVisitor::VisitLstmLayer().

m_CifgEnabled

bool m_CifgEnabled

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

Definition at line 867 of file Descriptors.hpp.

Referenced by armnn::BOOST_AUTO_TEST_CASE(), BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), LstmLayer::CreateWorkload(), RefLstmWorkload::Execute(), Deserializer::GetLstmDescriptor(), LstmLayer::InferOutputShapes(), RefLayerSupport::IsLstmSupported(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), LstmLayer::ValidateTensorShapesFromInputs(), and SerializerVisitor::VisitLstmLayer().

m_ClippingThresCell

float m_ClippingThresCell

Clipping threshold value for the cell state.

Definition at line 863 of file Descriptors.hpp.

Referenced by armnn::BOOST_AUTO_TEST_CASE(), BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), RefLstmWorkload::Execute(), Deserializer::GetLstmDescriptor(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), and SerializerVisitor::VisitLstmLayer().

m_ClippingThresProj

float m_ClippingThresProj

Clipping threshold value for the projection.

Definition at line 865 of file Descriptors.hpp.

Referenced by armnn::BOOST_AUTO_TEST_CASE(), BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), RefLstmWorkload::Execute(), Deserializer::GetLstmDescriptor(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), and SerializerVisitor::VisitLstmLayer().

m_LayerNormEnabled

bool m_LayerNormEnabled

Enable/disable layer normalization.

Definition at line 873 of file Descriptors.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), LstmLayer::CreateWorkload(), RefLstmWorkload::Execute(), Deserializer::GetLstmDescriptor(), RefLayerSupport::IsLstmSupported(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), LstmLayer::ValidateTensorShapesFromInputs(), and SerializerVisitor::VisitLstmLayer().

m_PeepholeEnabled

bool m_PeepholeEnabled

Enable/disable peephole.

Definition at line 869 of file Descriptors.hpp.

Referenced by armnn::BOOST_AUTO_TEST_CASE(), BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), LstmLayer::CreateWorkload(), RefLstmWorkload::Execute(), Deserializer::GetLstmDescriptor(), RefLayerSupport::IsLstmSupported(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), LstmLayer::ValidateTensorShapesFromInputs(), and SerializerVisitor::VisitLstmLayer().

m_ProjectionEnabled

bool m_ProjectionEnabled

Enable/disable the projection layer.

Definition at line 871 of file Descriptors.hpp.

Referenced by armnn::BOOST_AUTO_TEST_CASE(), BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), LstmLayer::CreateWorkload(), RefLstmWorkload::Execute(), Deserializer::GetLstmDescriptor(), RefLayerSupport::IsLstmSupported(), StringifyLayerParameters< LstmDescriptor >::Serialize(), LstmLayer::ValidateTensorShapesFromInputs(), and SerializerVisitor::VisitLstmLayer().

---

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

• include/armnn/Descriptors.hpp