24.02
|
#include <LstmParameters.hpp>
Public Attributes | |
std::shared_ptr< ConstTensorHandle > | m_InputLayerNormWeights |
A unique pointer to represent 1D weights tensor with dimensions [num_units]. More... | |
std::shared_ptr< ConstTensorHandle > | m_ForgetLayerNormWeights |
A unique pointer to represent 1D weights tensor with dimensions [num_units]. More... | |
std::shared_ptr< ConstTensorHandle > | m_CellLayerNormWeights |
A unique pointer to represent 1D weights tensor with dimensions [num_units]. More... | |
std::shared_ptr< ConstTensorHandle > | m_OutputLayerNormWeights |
A unique pointer to represent 1D weights tensor with dimensions [num_units]. More... | |
Definition at line 14 of file LstmParameters.hpp.
std::shared_ptr<ConstTensorHandle> m_CellLayerNormWeights |
A unique pointer to represent 1D weights tensor with dimensions [num_units].
Definition at line 21 of file LstmParameters.hpp.
Referenced by LstmLayer::Clone(), UnidirectionalSequenceLstmLayer::Clone(), LstmLayer::CreateWorkload(), UnidirectionalSequenceLstmLayer::CreateWorkload(), LstmLayer::ExecuteStrategy(), UnidirectionalSequenceLstmLayer::ExecuteStrategy(), LstmLayer::GetConstantTensorsByRef(), UnidirectionalSequenceLstmLayer::GetConstantTensorsByRef(), and UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs().
std::shared_ptr<ConstTensorHandle> m_ForgetLayerNormWeights |
A unique pointer to represent 1D weights tensor with dimensions [num_units].
Definition at line 19 of file LstmParameters.hpp.
Referenced by LstmLayer::Clone(), UnidirectionalSequenceLstmLayer::Clone(), LstmLayer::CreateWorkload(), UnidirectionalSequenceLstmLayer::CreateWorkload(), LstmLayer::ExecuteStrategy(), UnidirectionalSequenceLstmLayer::ExecuteStrategy(), LstmLayer::GetConstantTensorsByRef(), UnidirectionalSequenceLstmLayer::GetConstantTensorsByRef(), and UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs().
std::shared_ptr<ConstTensorHandle> m_InputLayerNormWeights |
A unique pointer to represent 1D weights tensor with dimensions [num_units].
Definition at line 17 of file LstmParameters.hpp.
Referenced by LstmLayer::Clone(), UnidirectionalSequenceLstmLayer::Clone(), LstmLayer::CreateWorkload(), UnidirectionalSequenceLstmLayer::CreateWorkload(), LstmLayer::ExecuteStrategy(), UnidirectionalSequenceLstmLayer::ExecuteStrategy(), LstmLayer::GetConstantTensorsByRef(), UnidirectionalSequenceLstmLayer::GetConstantTensorsByRef(), and UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs().
std::shared_ptr<ConstTensorHandle> m_OutputLayerNormWeights |
A unique pointer to represent 1D weights tensor with dimensions [num_units].
Definition at line 23 of file LstmParameters.hpp.
Referenced by LstmLayer::Clone(), UnidirectionalSequenceLstmLayer::Clone(), LstmLayer::CreateWorkload(), UnidirectionalSequenceLstmLayer::CreateWorkload(), LstmLayer::ExecuteStrategy(), UnidirectionalSequenceLstmLayer::ExecuteStrategy(), LstmLayer::GetConstantTensorsByRef(), UnidirectionalSequenceLstmLayer::GetConstantTensorsByRef(), and UnidirectionalSequenceLstmLayer::ValidateTensorShapesFromInputs().