ArmNN
NotReleased
|
#include <LstmLayer.hpp>
Public Attributes | |
std::unique_ptr< ScopedCpuTensorHandle > | m_InputToForgetWeights |
A unique pointer to represent 2D weights tensor with dimensions [input_size, num_units]. More... | |
std::unique_ptr< ScopedCpuTensorHandle > | m_InputToCellWeights |
A unique pointer to represent 2D weights tensor with dimensions [input_size, num_units]. More... | |
std::unique_ptr< ScopedCpuTensorHandle > | m_InputToOutputWeights |
A unique pointer to represent 2D weights tensor with dimensions [input_size, num_units]. More... | |
std::unique_ptr< ScopedCpuTensorHandle > | m_RecurrentToForgetWeights |
A unique pointer to represent 2D weights tensor with dimensions [output_size, num_units]. More... | |
std::unique_ptr< ScopedCpuTensorHandle > | m_RecurrentToCellWeights |
A unique pointer to represent 2D weights tensor with dimensions [output_size, num_units]. More... | |
std::unique_ptr< ScopedCpuTensorHandle > | m_RecurrentToOutputWeights |
A unique pointer to represent 2D weights tensor with dimensions [output_size, num_units]. More... | |
std::unique_ptr< ScopedCpuTensorHandle > | m_ForgetGateBias |
A unique pointer to represent 1D weights tensor with dimensions [num_units]. More... | |
std::unique_ptr< ScopedCpuTensorHandle > | m_CellBias |
A unique pointer to represent 1D weights tensor with dimensions [num_units]. More... | |
std::unique_ptr< ScopedCpuTensorHandle > | m_OutputGateBias |
A unique pointer to represent 1D weights tensor with dimensions [num_units]. More... | |
Definition at line 54 of file LstmLayer.hpp.
std::unique_ptr<ScopedCpuTensorHandle> m_CellBias |
A unique pointer to represent 1D weights tensor with dimensions [num_units].
Definition at line 71 of file LstmLayer.hpp.
Referenced by LstmLayer::Accept(), LstmLayer::Clone(), LstmLayer::CreateWorkload(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().
std::unique_ptr<ScopedCpuTensorHandle> m_ForgetGateBias |
A unique pointer to represent 1D weights tensor with dimensions [num_units].
Definition at line 69 of file LstmLayer.hpp.
Referenced by LstmLayer::Accept(), LstmLayer::Clone(), LstmLayer::CreateWorkload(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().
std::unique_ptr<ScopedCpuTensorHandle> m_InputToCellWeights |
A unique pointer to represent 2D weights tensor with dimensions [input_size, num_units].
Definition at line 59 of file LstmLayer.hpp.
Referenced by LstmLayer::Accept(), LstmLayer::Clone(), LstmLayer::CreateWorkload(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().
std::unique_ptr<ScopedCpuTensorHandle> m_InputToForgetWeights |
A unique pointer to represent 2D weights tensor with dimensions [input_size, num_units].
Definition at line 57 of file LstmLayer.hpp.
Referenced by LstmLayer::Accept(), Network::AddLstmLayer(), LstmLayer::Clone(), LstmLayer::CreateWorkload(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().
std::unique_ptr<ScopedCpuTensorHandle> m_InputToOutputWeights |
A unique pointer to represent 2D weights tensor with dimensions [input_size, num_units].
Definition at line 61 of file LstmLayer.hpp.
Referenced by LstmLayer::Accept(), LstmLayer::Clone(), LstmLayer::CreateWorkload(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().
std::unique_ptr<ScopedCpuTensorHandle> m_OutputGateBias |
A unique pointer to represent 1D weights tensor with dimensions [num_units].
Definition at line 73 of file LstmLayer.hpp.
Referenced by LstmLayer::Accept(), LstmLayer::Clone(), LstmLayer::CreateWorkload(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().
std::unique_ptr<ScopedCpuTensorHandle> m_RecurrentToCellWeights |
A unique pointer to represent 2D weights tensor with dimensions [output_size, num_units].
Definition at line 65 of file LstmLayer.hpp.
Referenced by LstmLayer::Accept(), LstmLayer::Clone(), LstmLayer::CreateWorkload(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().
std::unique_ptr<ScopedCpuTensorHandle> m_RecurrentToForgetWeights |
A unique pointer to represent 2D weights tensor with dimensions [output_size, num_units].
Definition at line 63 of file LstmLayer.hpp.
Referenced by LstmLayer::Accept(), LstmLayer::Clone(), LstmLayer::CreateWorkload(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().
std::unique_ptr<ScopedCpuTensorHandle> m_RecurrentToOutputWeights |
A unique pointer to represent 2D weights tensor with dimensions [output_size, num_units].
Definition at line 67 of file LstmLayer.hpp.
Referenced by LstmLayer::Accept(), LstmLayer::Clone(), LstmLayer::CreateWorkload(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().