20.02
|
#include <LstmLayer.hpp>
Public Attributes | |
std::unique_ptr< ScopedCpuTensorHandle > | m_InputToInputWeights |
A unique pointer to represent 2D weights tensor with dimensions [input_size, num_units]. More... | |
std::unique_ptr< ScopedCpuTensorHandle > | m_RecurrentToInputWeights |
A unique pointer to represent 2D weights tensor with dimensions [input_size, num_units]. More... | |
std::unique_ptr< ScopedCpuTensorHandle > | m_CellToInputWeights |
A unique pointer to represent 1D weights tensor with dimensions [num_units]. More... | |
std::unique_ptr< ScopedCpuTensorHandle > | m_InputGateBias |
A unique pointer to represent 1D weights tensor with dimensions [num_units]. More... | |
Definition at line 26 of file LstmLayer.hpp.
std::unique_ptr<ScopedCpuTensorHandle> m_CellToInputWeights |
A unique pointer to represent 1D weights tensor with dimensions [num_units].
Definition at line 33 of file LstmLayer.hpp.
Referenced by LstmLayer::Accept(), LstmLayer::Clone(), LstmLayer::CreateWorkload(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().
std::unique_ptr<ScopedCpuTensorHandle> m_InputGateBias |
A unique pointer to represent 1D weights tensor with dimensions [num_units].
Definition at line 35 of file LstmLayer.hpp.
Referenced by LstmLayer::Accept(), LstmLayer::Clone(), LstmLayer::CreateWorkload(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().
std::unique_ptr<ScopedCpuTensorHandle> m_InputToInputWeights |
A unique pointer to represent 2D weights tensor with dimensions [input_size, num_units].
Definition at line 29 of file LstmLayer.hpp.
Referenced by LstmLayer::Accept(), LstmLayer::Clone(), LstmLayer::CreateWorkload(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().
std::unique_ptr<ScopedCpuTensorHandle> m_RecurrentToInputWeights |
A unique pointer to represent 2D weights tensor with dimensions [input_size, num_units].
Definition at line 31 of file LstmLayer.hpp.
Referenced by LstmLayer::Accept(), LstmLayer::Clone(), LstmLayer::CreateWorkload(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().