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

#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...
 

Detailed Description

Definition at line 54 of file LstmLayer.hpp.

Member Data Documentation

◆ m_CellBias

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

◆ m_ForgetGateBias

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

◆ m_InputToCellWeights

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

◆ m_InputToForgetWeights

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

◆ m_InputToOutputWeights

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

◆ m_OutputGateBias

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

◆ m_RecurrentToCellWeights

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

◆ m_RecurrentToForgetWeights

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

◆ m_RecurrentToOutputWeights

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

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