21.08
|
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... | |
bool | m_TimeMajor |
Enable/disable time major. More... | |
An LstmDescriptor for the LstmLayer.
Definition at line 923 of file Descriptors.hpp.
|
inline |
Definition at line 925 of file Descriptors.hpp.
|
inline |
Definition at line 936 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 949 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(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), and UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest().
bool m_CifgEnabled |
Enable/disable cifg (coupled input & forget gate).
Definition at line 955 of file Descriptors.hpp.
Referenced by TestLstmLayerVisitor::CheckDescriptor(), LstmLayer::CreateWorkload(), UnidirectionalSequenceLstmLayer::CreateWorkload(), RefLstmWorkload::ExecuteAsync(), RefUnidirectionalSequenceLstmWorkload::ExecuteAsync(), UnidirectionalSequenceLstmLayer::ExecuteStrategy(), LstmLayer::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(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest(), LstmLayer::ValidateTensorShapesFromInputs(), and UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs().
float m_ClippingThresCell |
Clipping threshold value for the cell state.
Definition at line 951 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(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), and UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest().
float m_ClippingThresProj |
Clipping threshold value for the projection.
Definition at line 953 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(), and UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest().
bool m_LayerNormEnabled |
Enable/disable layer normalization.
Definition at line 961 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(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), and UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs().
bool m_PeepholeEnabled |
Enable/disable peephole.
Definition at line 957 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(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest(), LstmLayer::ValidateTensorShapesFromInputs(), and UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs().
bool m_ProjectionEnabled |
Enable/disable the projection layer.
Definition at line 959 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(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest(), UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs(), and LstmLayer::ValidateTensorShapesFromInputs().
bool m_TimeMajor |
Enable/disable time major.
Definition at line 963 of file Descriptors.hpp.
Referenced by RefUnidirectionalSequenceLstmWorkload::ExecuteAsync(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), UnidirectionalSequenceLstmLayer::InferOutputShapes(), LstmDescriptor::operator==(), TEST_SUITE(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), and UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest().