ArmNN
 25.11
Loading...
Searching...
No Matches
LstmBasicParameters Struct Reference

#include <LstmParameters.hpp>

Public Attributes

std::shared_ptr< ConstTensorHandlem_InputToForgetWeights
 A unique pointer to represent 2D weights tensor with dimensions [input_size, num_units].
std::shared_ptr< ConstTensorHandlem_InputToCellWeights
 A unique pointer to represent 2D weights tensor with dimensions [input_size, num_units].
std::shared_ptr< ConstTensorHandlem_InputToOutputWeights
 A unique pointer to represent 2D weights tensor with dimensions [input_size, num_units].
std::shared_ptr< ConstTensorHandlem_RecurrentToForgetWeights
 A unique pointer to represent 2D weights tensor with dimensions [output_size, num_units].
std::shared_ptr< ConstTensorHandlem_RecurrentToCellWeights
 A unique pointer to represent 2D weights tensor with dimensions [output_size, num_units].
std::shared_ptr< ConstTensorHandlem_RecurrentToOutputWeights
 A unique pointer to represent 2D weights tensor with dimensions [output_size, num_units].
std::shared_ptr< ConstTensorHandlem_ForgetGateBias
 A unique pointer to represent 1D weights tensor with dimensions [num_units].
std::shared_ptr< ConstTensorHandlem_CellBias
 A unique pointer to represent 1D weights tensor with dimensions [num_units].
std::shared_ptr< ConstTensorHandlem_OutputGateBias
 A unique pointer to represent 1D weights tensor with dimensions [num_units].

Detailed Description

Definition at line 54 of file LstmParameters.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 LstmParameters.hpp.

◆ 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 LstmParameters.hpp.

◆ 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 LstmParameters.hpp.

◆ 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 LstmParameters.hpp.

Referenced by NetworkImpl::AddLstmLayer(), and NetworkImpl::AddUnidirectionalSequenceLstmLayer().

◆ 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 LstmParameters.hpp.

◆ 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 LstmParameters.hpp.

◆ 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 LstmParameters.hpp.

◆ 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 LstmParameters.hpp.

◆ 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 LstmParameters.hpp.


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