21.05
|
#include <LstmLayer.hpp>
Public Attributes | |
std::shared_ptr< ConstTensorHandle > | m_CellToInputWeights |
A unique pointer to represent 1D weights tensor with dimensions [num_units]. More... | |
std::shared_ptr< ConstTensorHandle > | m_CellToForgetWeights |
A unique pointer to represent 1D weights tensor with dimensions [num_units]. More... | |
std::shared_ptr< ConstTensorHandle > | m_CellToOutputWeights |
A unique pointer to represent 1D weights tensor with dimensions [num_units]. More... | |
Definition at line 44 of file LstmLayer.hpp.
std::shared_ptr<ConstTensorHandle> m_CellToForgetWeights |
A unique pointer to represent 1D weights tensor with dimensions [num_units].
Definition at line 49 of file LstmLayer.hpp.
Referenced by LstmLayer::Accept(), LstmLayer::Clone(), LstmLayer::CreateWorkload(), LstmLayer::ExecuteStrategy(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().
std::shared_ptr<ConstTensorHandle> m_CellToInputWeights |
A unique pointer to represent 1D weights tensor with dimensions [num_units].
Definition at line 47 of file LstmLayer.hpp.
Referenced by LstmLayer::Accept(), LstmLayer::Clone(), LstmLayer::CreateWorkload(), LstmLayer::ExecuteStrategy(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().
std::shared_ptr<ConstTensorHandle> m_CellToOutputWeights |
A unique pointer to represent 1D weights tensor with dimensions [num_units].
Definition at line 51 of file LstmLayer.hpp.
Referenced by LstmLayer::Accept(), LstmLayer::Clone(), LstmLayer::CreateWorkload(), LstmLayer::ExecuteStrategy(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().