20.02
|
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 837 of file Descriptors.hpp.
|
inline |
Definition at line 839 of file Descriptors.hpp.
|
inline |
Definition at line 849 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 861 of file Descriptors.hpp.
Referenced by armnn::BOOST_AUTO_TEST_CASE(), BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), RefLstmWorkload::Execute(), Deserializer::GetLstmDescriptor(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), and SerializerVisitor::VisitLstmLayer().
bool m_CifgEnabled |
Enable/disable cifg (coupled input & forget gate).
Definition at line 867 of file Descriptors.hpp.
Referenced by armnn::BOOST_AUTO_TEST_CASE(), BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), LstmLayer::CreateWorkload(), RefLstmWorkload::Execute(), Deserializer::GetLstmDescriptor(), LstmLayer::InferOutputShapes(), RefLayerSupport::IsLstmSupported(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), LstmLayer::ValidateTensorShapesFromInputs(), and SerializerVisitor::VisitLstmLayer().
float m_ClippingThresCell |
Clipping threshold value for the cell state.
Definition at line 863 of file Descriptors.hpp.
Referenced by armnn::BOOST_AUTO_TEST_CASE(), BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), RefLstmWorkload::Execute(), Deserializer::GetLstmDescriptor(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), and SerializerVisitor::VisitLstmLayer().
float m_ClippingThresProj |
Clipping threshold value for the projection.
Definition at line 865 of file Descriptors.hpp.
Referenced by armnn::BOOST_AUTO_TEST_CASE(), BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), RefLstmWorkload::Execute(), Deserializer::GetLstmDescriptor(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), and SerializerVisitor::VisitLstmLayer().
bool m_LayerNormEnabled |
Enable/disable layer normalization.
Definition at line 873 of file Descriptors.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), LstmLayer::CreateWorkload(), RefLstmWorkload::Execute(), Deserializer::GetLstmDescriptor(), RefLayerSupport::IsLstmSupported(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), LstmLayer::ValidateTensorShapesFromInputs(), and SerializerVisitor::VisitLstmLayer().
bool m_PeepholeEnabled |
Enable/disable peephole.
Definition at line 869 of file Descriptors.hpp.
Referenced by armnn::BOOST_AUTO_TEST_CASE(), BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), LstmLayer::CreateWorkload(), RefLstmWorkload::Execute(), Deserializer::GetLstmDescriptor(), RefLayerSupport::IsLstmSupported(), LstmDescriptor::operator==(), StringifyLayerParameters< LstmDescriptor >::Serialize(), LstmLayer::ValidateTensorShapesFromInputs(), and SerializerVisitor::VisitLstmLayer().
bool m_ProjectionEnabled |
Enable/disable the projection layer.
Definition at line 871 of file Descriptors.hpp.
Referenced by armnn::BOOST_AUTO_TEST_CASE(), BOOST_AUTO_TEST_CASE(), TestLstmLayerVisitor::CheckDescriptor(), LstmLayer::CreateWorkload(), RefLstmWorkload::Execute(), Deserializer::GetLstmDescriptor(), RefLayerSupport::IsLstmSupported(), StringifyLayerParameters< LstmDescriptor >::Serialize(), LstmLayer::ValidateTensorShapesFromInputs(), and SerializerVisitor::VisitLstmLayer().