24.08
|
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... | |
float | m_InputIntermediateScale |
Input intermediate quantization scale. More... | |
float | m_ForgetIntermediateScale |
Forget intermediate quantization scale. More... | |
float | m_CellIntermediateScale |
Cell intermediate quantization scale. More... | |
float | m_OutputIntermediateScale |
Output intermediate quantization scale. More... | |
int32_t | m_HiddenStateZeroPoint |
Hidden State zero point. More... | |
float | m_HiddenStateScale |
Hidden State quantization scale. More... | |
An LstmDescriptor for the LstmLayer.
Definition at line 1102 of file Descriptors.hpp.
|
inline |
Definition at line 1104 of file Descriptors.hpp.
|
inline |
Definition at line 1121 of file Descriptors.hpp.
References LstmDescriptor::m_ActivationFunc, LstmDescriptor::m_CellIntermediateScale, LstmDescriptor::m_CifgEnabled, LstmDescriptor::m_ClippingThresCell, LstmDescriptor::m_ClippingThresProj, LstmDescriptor::m_ForgetIntermediateScale, LstmDescriptor::m_HiddenStateScale, LstmDescriptor::m_HiddenStateZeroPoint, LstmDescriptor::m_InputIntermediateScale, LstmDescriptor::m_LayerNormEnabled, LstmDescriptor::m_OutputIntermediateScale, 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 1140 of file Descriptors.hpp.
Referenced by IDeserializer::DeserializerImpl::GetLstmDescriptor(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), armnn::LstmImpl(), LstmDescriptor::operator==(), and StringifyLayerParameters< LstmDescriptor >::Serialize().
float m_CellIntermediateScale |
Cell intermediate quantization scale.
Definition at line 1160 of file Descriptors.hpp.
Referenced by LstmDescriptor::operator==().
bool m_CifgEnabled |
Enable/disable cifg (coupled input & forget gate).
Definition at line 1146 of file Descriptors.hpp.
Referenced by LstmLayer::Clone(), UnidirectionalSequenceLstmLayer::Clone(), LstmLayer::CreateWorkload(), UnidirectionalSequenceLstmLayer::CreateWorkload(), LstmLayer::ExecuteStrategy(), UnidirectionalSequenceLstmLayer::ExecuteStrategy(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), LstmLayer::InferOutputShapes(), RefLayerSupport::IsLstmSupported(), RefLayerSupport::IsUnidirectionalSequenceLstmSupported(), armnn::LstmImpl(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), LstmQueueDescriptor::Validate(), UnidirectionalSequenceLstmQueueDescriptor::Validate(), UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs(), and LstmLayer::ValidateTensorShapesFromInputs().
float m_ClippingThresCell |
Clipping threshold value for the cell state.
Definition at line 1142 of file Descriptors.hpp.
Referenced by IDeserializer::DeserializerImpl::GetLstmDescriptor(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), armnn::LstmImpl(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), LstmQueueDescriptor::Validate(), and UnidirectionalSequenceLstmQueueDescriptor::Validate().
float m_ClippingThresProj |
Clipping threshold value for the projection.
Definition at line 1144 of file Descriptors.hpp.
Referenced by IDeserializer::DeserializerImpl::GetLstmDescriptor(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), armnn::LstmImpl(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), LstmQueueDescriptor::Validate(), and UnidirectionalSequenceLstmQueueDescriptor::Validate().
float m_ForgetIntermediateScale |
Forget intermediate quantization scale.
Definition at line 1158 of file Descriptors.hpp.
Referenced by LstmDescriptor::operator==().
float m_HiddenStateScale |
Hidden State quantization scale.
Definition at line 1166 of file Descriptors.hpp.
Referenced by LstmDescriptor::operator==().
int32_t m_HiddenStateZeroPoint |
Hidden State zero point.
Definition at line 1164 of file Descriptors.hpp.
Referenced by LstmDescriptor::operator==().
float m_InputIntermediateScale |
Input intermediate quantization scale.
Definition at line 1156 of file Descriptors.hpp.
Referenced by LstmDescriptor::operator==().
bool m_LayerNormEnabled |
Enable/disable layer normalization.
Definition at line 1152 of file Descriptors.hpp.
Referenced by LstmLayer::Clone(), UnidirectionalSequenceLstmLayer::Clone(), UnidirectionalSequenceLstmLayer::CreateWorkload(), LstmLayer::CreateWorkload(), LstmLayer::ExecuteStrategy(), UnidirectionalSequenceLstmLayer::ExecuteStrategy(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), RefLayerSupport::IsLstmSupported(), RefLayerSupport::IsUnidirectionalSequenceLstmSupported(), armnn::LstmImpl(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), LstmQueueDescriptor::Validate(), UnidirectionalSequenceLstmQueueDescriptor::Validate(), and UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs().
float m_OutputIntermediateScale |
Output intermediate quantization scale.
Definition at line 1162 of file Descriptors.hpp.
Referenced by LstmDescriptor::operator==().
bool m_PeepholeEnabled |
Enable/disable peephole.
Definition at line 1148 of file Descriptors.hpp.
Referenced by LstmLayer::Clone(), UnidirectionalSequenceLstmLayer::Clone(), UnidirectionalSequenceLstmLayer::CreateWorkload(), LstmLayer::CreateWorkload(), LstmLayer::ExecuteStrategy(), UnidirectionalSequenceLstmLayer::ExecuteStrategy(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), RefLayerSupport::IsLstmSupported(), RefLayerSupport::IsUnidirectionalSequenceLstmSupported(), armnn::LstmImpl(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), LstmQueueDescriptor::Validate(), UnidirectionalSequenceLstmQueueDescriptor::Validate(), UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs(), and LstmLayer::ValidateTensorShapesFromInputs().
bool m_ProjectionEnabled |
Enable/disable the projection layer.
Definition at line 1150 of file Descriptors.hpp.
Referenced by LstmLayer::Clone(), UnidirectionalSequenceLstmLayer::Clone(), LstmLayer::CreateWorkload(), UnidirectionalSequenceLstmLayer::CreateWorkload(), LstmLayer::ExecuteStrategy(), UnidirectionalSequenceLstmLayer::ExecuteStrategy(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), RefLayerSupport::IsLstmSupported(), RefLayerSupport::IsUnidirectionalSequenceLstmSupported(), armnn::LstmImpl(), StringifyLayerParameters< LstmDescriptor >::Serialize(), LstmQueueDescriptor::Validate(), UnidirectionalSequenceLstmQueueDescriptor::Validate(), LstmLayer::ValidateTensorShapesFromInputs(), and UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs().
bool m_TimeMajor |
Enable/disable time major.
Definition at line 1154 of file Descriptors.hpp.
Referenced by armnn::ClUnidirectionalSequenceLstmFloatWorkloadValidate(), IDeserializer::DeserializerImpl::GetUnidirectionalSequenceLstmDescriptor(), UnidirectionalSequenceLstmLayer::InferOutputShapes(), armnn::NeonUnidirectionalSequenceLstmFloatWorkloadValidate(), armnn::NeonUnidirectionalSequenceLstmWorkloadValidate(), LstmDescriptor::operator==(), and UnidirectionalSequenceLstmQueueDescriptor::Validate().