ArmNN
 21.05
Public Attributes | List of all members
LstmBasicParameters Struct Reference

#include <LstmLayer.hpp>

Public Attributes

std::shared_ptr< ConstTensorHandle > m_InputToForgetWeights
 A unique pointer to represent 2D weights tensor with dimensions [input_size, num_units]. More...
 
std::shared_ptr< ConstTensorHandle > m_InputToCellWeights
 A unique pointer to represent 2D weights tensor with dimensions [input_size, num_units]. More...
 
std::shared_ptr< ConstTensorHandle > m_InputToOutputWeights
 A unique pointer to represent 2D weights tensor with dimensions [input_size, num_units]. More...
 
std::shared_ptr< ConstTensorHandle > m_RecurrentToForgetWeights
 A unique pointer to represent 2D weights tensor with dimensions [output_size, num_units]. More...
 
std::shared_ptr< ConstTensorHandle > m_RecurrentToCellWeights
 A unique pointer to represent 2D weights tensor with dimensions [output_size, num_units]. More...
 
std::shared_ptr< ConstTensorHandle > m_RecurrentToOutputWeights
 A unique pointer to represent 2D weights tensor with dimensions [output_size, num_units]. More...
 
std::shared_ptr< ConstTensorHandle > m_ForgetGateBias
 A unique pointer to represent 1D weights tensor with dimensions [num_units]. More...
 
std::shared_ptr< ConstTensorHandle > m_CellBias
 A unique pointer to represent 1D weights tensor with dimensions [num_units]. More...
 
std::shared_ptr< ConstTensorHandle > m_OutputGateBias
 A unique pointer to represent 1D weights tensor with dimensions [num_units]. More...
 

Detailed Description

Definition at line 54 of file LstmLayer.hpp.

Member Data Documentation

◆ m_CellBias

std::shared_ptr<ConstTensorHandle> 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::ExecuteStrategy(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().

◆ m_ForgetGateBias

std::shared_ptr<ConstTensorHandle> 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::ExecuteStrategy(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().

◆ m_InputToCellWeights

std::shared_ptr<ConstTensorHandle> 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::ExecuteStrategy(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().

◆ m_InputToForgetWeights

std::shared_ptr<ConstTensorHandle> 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(), NetworkImpl::AddLstmLayer(), LstmLayer::Clone(), LstmLayer::CreateWorkload(), LstmLayer::ExecuteStrategy(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().

◆ m_InputToOutputWeights

std::shared_ptr<ConstTensorHandle> 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::ExecuteStrategy(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().

◆ m_OutputGateBias

std::shared_ptr<ConstTensorHandle> 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::ExecuteStrategy(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().

◆ m_RecurrentToCellWeights

std::shared_ptr<ConstTensorHandle> 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::ExecuteStrategy(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().

◆ m_RecurrentToForgetWeights

std::shared_ptr<ConstTensorHandle> 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::ExecuteStrategy(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().

◆ m_RecurrentToOutputWeights

std::shared_ptr<ConstTensorHandle> 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::ExecuteStrategy(), LstmLayer::GetConstantTensorsByRef(), and LstmLayer::ValidateTensorShapesFromInputs().

The documentation for this struct was generated from the following file: