21.05
|
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... | |
An LstmDescriptor for the LstmLayer.
Definition at line 919 of file Descriptors.hpp.
|
inline |
Definition at line 921 of file Descriptors.hpp.
|
inline |
Definition at line 931 of file Descriptors.hpp.
References LstmDescriptor::m_ActivationFunc, LstmDescriptor::m_CifgEnabled, LstmDescriptor::m_ClippingThresCell, LstmDescriptor::m_ClippingThresProj, LstmDescriptor::m_LayerNormEnabled, and LstmDescriptor::m_PeepholeEnabled.
uint32_t m_ActivationFunc |
The activation function to use.
0: None, 1: Relu, 3: Relu6, 4: Tanh, 6: Sigmoid.
Definition at line 943 of file Descriptors.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), RefLstmWorkload::ExecuteAsync(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), LstmDescriptor::operator==(), and StringifyLayerParameters< LstmDescriptor >::Serialize().
bool m_CifgEnabled |
Enable/disable cifg (coupled input & forget gate).
Definition at line 949 of file Descriptors.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), LstmLayer::CreateWorkload(), RefLstmWorkload::ExecuteAsync(), LstmLayer::ExecuteStrategy(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), LstmLayer::InferOutputShapes(), RefLayerSupport::IsLstmSupported(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), and LstmLayer::ValidateTensorShapesFromInputs().
float m_ClippingThresCell |
Clipping threshold value for the cell state.
Definition at line 945 of file Descriptors.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), RefLstmWorkload::ExecuteAsync(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), LstmDescriptor::operator==(), and StringifyLayerParameters< LstmDescriptor >::Serialize().
float m_ClippingThresProj |
Clipping threshold value for the projection.
Definition at line 947 of file Descriptors.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), RefLstmWorkload::ExecuteAsync(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), LstmDescriptor::operator==(), and StringifyLayerParameters< LstmDescriptor >::Serialize().
bool m_LayerNormEnabled |
Enable/disable layer normalization.
Definition at line 955 of file Descriptors.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), LstmLayer::CreateWorkload(), RefLstmWorkload::ExecuteAsync(), LstmLayer::ExecuteStrategy(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), RefLayerSupport::IsLstmSupported(), LstmDescriptor::operator==(), and StringifyLayerParameters< LstmDescriptor >::Serialize().
bool m_PeepholeEnabled |
Enable/disable peephole.
Definition at line 951 of file Descriptors.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), LstmLayer::CreateWorkload(), RefLstmWorkload::ExecuteAsync(), LstmLayer::ExecuteStrategy(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), RefLayerSupport::IsLstmSupported(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), and LstmLayer::ValidateTensorShapesFromInputs().
bool m_ProjectionEnabled |
Enable/disable the projection layer.
Definition at line 953 of file Descriptors.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), armnn::BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), LstmLayer::CreateWorkload(), RefLstmWorkload::ExecuteAsync(), LstmLayer::ExecuteStrategy(), armnnSerializer::GetFlatBufferArgMinMaxFunction(), IDeserializer::DeserializerImpl::GetLstmDescriptor(), RefLayerSupport::IsLstmSupported(), StringifyLayerParameters< LstmDescriptor >::Serialize(), and LstmLayer::ValidateTensorShapesFromInputs().