22.02
|
An LstmDescriptor for the LstmLayer. More...
#include <Descriptors.hpp>
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... | |
An LstmDescriptor for the LstmLayer.
Definition at line 1078 of file Descriptors.hpp.
|
inline |
Definition at line 1080 of file Descriptors.hpp.
|
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.
uint32_t m_ActivationFunc |
The activation function to use.
0: None, 1: Relu, 3: Relu6, 4: Tanh, 6: Sigmoid.
Definition at line 1104 of file Descriptors.hpp.
Referenced by TestLstmLayerVisitor::CheckDescriptor(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), armnn::LstmImpl(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), TEST_SUITE(), armnn::TEST_SUITE(), UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjectionTest(), UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest(), UnidirectionalSequenceLstmLayerInt8Test(), UnidirectionalSequenceLstmLayerInt8TimeMajorTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), and UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest().
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().
float m_ClippingThresCell |
Clipping threshold value for the cell state.
Definition at line 1106 of file Descriptors.hpp.
Referenced by TestLstmLayerVisitor::CheckDescriptor(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), armnn::LstmImpl(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), TEST_SUITE(), armnn::TEST_SUITE(), UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjectionTest(), UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest(), UnidirectionalSequenceLstmLayerInt8Test(), UnidirectionalSequenceLstmLayerInt8TimeMajorTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), and UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest().
float m_ClippingThresProj |
Clipping threshold value for the projection.
Definition at line 1108 of file Descriptors.hpp.
Referenced by TestLstmLayerVisitor::CheckDescriptor(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), armnn::LstmImpl(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), TEST_SUITE(), armnn::TEST_SUITE(), UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjectionTest(), UnidirectionalSequenceLstmLayerInt8Test(), UnidirectionalSequenceLstmLayerInt8TimeMajorTest(), and UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest().
bool m_LayerNormEnabled |
Enable/disable layer normalization.
Definition at line 1116 of file Descriptors.hpp.
Referenced by LstmLayer::CreateWorkload(), UnidirectionalSequenceLstmLayer::CreateWorkload(), RefLstmWorkload::ExecuteAsync(), RefUnidirectionalSequenceLstmWorkload::ExecuteAsync(), LstmLayer::ExecuteStrategy(), UnidirectionalSequenceLstmLayer::ExecuteStrategy(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), RefLayerSupport::IsLstmSupported(), RefLayerSupport::IsUnidirectionalSequenceLstmSupported(), armnn::LstmImpl(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), TEST_SUITE(), UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), and UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs().
bool m_PeepholeEnabled |
Enable/disable peephole.
Definition at line 1112 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(), RefLayerSupport::IsLstmSupported(), RefLayerSupport::IsUnidirectionalSequenceLstmSupported(), armnn::LstmImpl(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), TEST_SUITE(), armnn::TEST_SUITE(), UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjectionTest(), UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest(), UnidirectionalSequenceLstmLayerInt8Test(), UnidirectionalSequenceLstmLayerInt8TimeMajorTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest(), UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs(), and LstmLayer::ValidateTensorShapesFromInputs().
bool m_ProjectionEnabled |
Enable/disable the projection layer.
Definition at line 1114 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(), RefLayerSupport::IsLstmSupported(), RefLayerSupport::IsUnidirectionalSequenceLstmSupported(), armnn::LstmImpl(), StringifyLayerParameters< LstmDescriptor >::Serialize(), TEST_SUITE(), armnn::TEST_SUITE(), UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjectionTest(), UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest(), UnidirectionalSequenceLstmLayerInt8Test(), UnidirectionalSequenceLstmLayerInt8TimeMajorTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest(), UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs(), and LstmLayer::ValidateTensorShapesFromInputs().
bool m_TimeMajor |
Enable/disable time major.
Definition at line 1118 of file Descriptors.hpp.
Referenced by RefUnidirectionalSequenceLstmWorkload::ExecuteAsync(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), UnidirectionalSequenceLstmLayer::InferOutputShapes(), LstmDescriptor::operator==(), TEST_SUITE(), UnidirectionalSequenceLstmInt8WithCifgWithPeepholeNoProjectionTest(), UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerInt8NoCifgWithPeepholeWithProjectionWithLayerNormTest(), UnidirectionalSequenceLstmLayerInt8Test(), UnidirectionalSequenceLstmLayerInt8TimeMajorTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), and UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest().