LstmDescriptor Struct Reference

An LstmDescriptor for the LstmLayer. More...

#include <Descriptors.hpp>

Inheritance diagram for LstmDescriptor:

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

Constructor & Destructor Documentation

LstmDescriptor()

LstmDescriptor ( )

inline

Definition at line 913 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 923 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 935 of file Descriptors.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), RefLstmWorkload::Execute(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), LstmDescriptor::operator==(), and StringifyLayerParameters< LstmDescriptor >::Serialize().

m_CifgEnabled

bool m_CifgEnabled

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

Definition at line 941 of file Descriptors.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), LstmLayer::CreateWorkload(), RefLstmWorkload::Execute(), LstmLayer::ExecuteStrategy(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), LstmLayer::InferOutputShapes(), RefLayerSupport::IsLstmSupported(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), and LstmLayer::ValidateTensorShapesFromInputs().

m_ClippingThresCell

float m_ClippingThresCell

Clipping threshold value for the cell state.

Definition at line 937 of file Descriptors.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), RefLstmWorkload::Execute(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), LstmDescriptor::operator==(), and StringifyLayerParameters< LstmDescriptor >::Serialize().

m_ClippingThresProj

float m_ClippingThresProj

Clipping threshold value for the projection.

Definition at line 939 of file Descriptors.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), RefLstmWorkload::Execute(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), LstmDescriptor::operator==(), and StringifyLayerParameters< LstmDescriptor >::Serialize().

m_LayerNormEnabled

bool m_LayerNormEnabled

Enable/disable layer normalization.

Definition at line 947 of file Descriptors.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), LstmLayer::CreateWorkload(), RefLstmWorkload::Execute(), LstmLayer::ExecuteStrategy(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), RefLayerSupport::IsLstmSupported(), LstmDescriptor::operator==(), and StringifyLayerParameters< LstmDescriptor >::Serialize().

m_PeepholeEnabled

bool m_PeepholeEnabled

Enable/disable peephole.

Definition at line 943 of file Descriptors.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), LstmLayer::CreateWorkload(), RefLstmWorkload::Execute(), LstmLayer::ExecuteStrategy(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), RefLayerSupport::IsLstmSupported(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), and LstmLayer::ValidateTensorShapesFromInputs().

m_ProjectionEnabled

bool m_ProjectionEnabled

Enable/disable the projection layer.

Definition at line 945 of file Descriptors.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), LstmLayer::CreateWorkload(), RefLstmWorkload::Execute(), LstmLayer::ExecuteStrategy(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), RefLayerSupport::IsLstmSupported(), StringifyLayerParameters< LstmDescriptor >::Serialize(), and LstmLayer::ValidateTensorShapesFromInputs().

---

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

• include/armnn/Descriptors.hpp