ArmNN
 26.07
LstmInputParamsInfo Struct Reference

#include <LstmParams.hpp>

Collaboration diagram for LstmInputParamsInfo:
[legend]

Public Member Functions

 LstmInputParamsInfo ()
 
const TensorInfo & Deref (const TensorInfo *tensorInfo) const
 
const TensorInfo & GetInputToInputWeights () const
 
const TensorInfo & GetInputToForgetWeights () const
 
const TensorInfo & GetInputToCellWeights () const
 
const TensorInfo & GetInputToOutputWeights () const
 
const TensorInfo & GetRecurrentToInputWeights () const
 
const TensorInfo & GetRecurrentToForgetWeights () const
 
const TensorInfo & GetRecurrentToCellWeights () const
 
const TensorInfo & GetRecurrentToOutputWeights () const
 
const TensorInfo & GetCellToInputWeights () const
 
const TensorInfo & GetCellToForgetWeights () const
 
const TensorInfo & GetCellToOutputWeights () const
 
const TensorInfo & GetInputGateBias () const
 
const TensorInfo & GetForgetGateBias () const
 
const TensorInfo & GetCellBias () const
 
const TensorInfo & GetOutputGateBias () const
 
const TensorInfo & GetProjectionWeights () const
 
const TensorInfo & GetProjectionBias () const
 
const TensorInfo & GetInputLayerNormWeights () const
 
const TensorInfo & GetForgetLayerNormWeights () const
 
const TensorInfo & GetCellLayerNormWeights () const
 
const TensorInfo & GetOutputLayerNormWeights () const
 

Public Attributes

const TensorInfo * m_InputToInputWeights
 
const TensorInfo * m_InputToForgetWeights
 
const TensorInfo * m_InputToCellWeights
 
const TensorInfo * m_InputToOutputWeights
 
const TensorInfo * m_RecurrentToInputWeights
 
const TensorInfo * m_RecurrentToForgetWeights
 
const TensorInfo * m_RecurrentToCellWeights
 
const TensorInfo * m_RecurrentToOutputWeights
 
const TensorInfo * m_CellToInputWeights
 
const TensorInfo * m_CellToForgetWeights
 
const TensorInfo * m_CellToOutputWeights
 
const TensorInfo * m_InputGateBias
 
const TensorInfo * m_ForgetGateBias
 
const TensorInfo * m_CellBias
 
const TensorInfo * m_OutputGateBias
 
const TensorInfo * m_ProjectionWeights
 
const TensorInfo * m_ProjectionBias
 
const TensorInfo * m_InputLayerNormWeights
 
const TensorInfo * m_ForgetLayerNormWeights
 
const TensorInfo * m_CellLayerNormWeights
 
const TensorInfo * m_OutputLayerNormWeights
 

Detailed Description

Definition at line 63 of file LstmParams.hpp.

Constructor & Destructor Documentation

◆ LstmInputParamsInfo()

LstmInputParamsInfo ( )
inline

Definition at line 65 of file LstmParams.hpp.

66  : m_InputToInputWeights(nullptr)
67  , m_InputToForgetWeights(nullptr)
68  , m_InputToCellWeights(nullptr)
69  , m_InputToOutputWeights(nullptr)
70  , m_RecurrentToInputWeights(nullptr)
72  , m_RecurrentToCellWeights(nullptr)
74  , m_CellToInputWeights(nullptr)
75  , m_CellToForgetWeights(nullptr)
76  , m_CellToOutputWeights(nullptr)
77  , m_InputGateBias(nullptr)
78  , m_ForgetGateBias(nullptr)
79  , m_CellBias(nullptr)
80  , m_OutputGateBias(nullptr)
81  , m_ProjectionWeights(nullptr)
82  , m_ProjectionBias(nullptr)
83  , m_InputLayerNormWeights(nullptr)
84  , m_ForgetLayerNormWeights(nullptr)
85  , m_CellLayerNormWeights(nullptr)
86  , m_OutputLayerNormWeights(nullptr)
87  {
88  }
const TensorInfo * m_InputGateBias
Definition: LstmParams.hpp:100
const TensorInfo * m_RecurrentToForgetWeights
Definition: LstmParams.hpp:94
const TensorInfo * m_InputToInputWeights
Definition: LstmParams.hpp:89
const TensorInfo * m_CellToOutputWeights
Definition: LstmParams.hpp:99
const TensorInfo * m_CellToForgetWeights
Definition: LstmParams.hpp:98
const TensorInfo * m_ForgetGateBias
Definition: LstmParams.hpp:101
const TensorInfo * m_InputToOutputWeights
Definition: LstmParams.hpp:92
const TensorInfo * m_OutputGateBias
Definition: LstmParams.hpp:103
const TensorInfo * m_InputToForgetWeights
Definition: LstmParams.hpp:90
const TensorInfo * m_ForgetLayerNormWeights
Definition: LstmParams.hpp:107
const TensorInfo * m_CellLayerNormWeights
Definition: LstmParams.hpp:108
const TensorInfo * m_RecurrentToCellWeights
Definition: LstmParams.hpp:95
const TensorInfo * m_OutputLayerNormWeights
Definition: LstmParams.hpp:109
const TensorInfo * m_RecurrentToInputWeights
Definition: LstmParams.hpp:93
const TensorInfo * m_CellBias
Definition: LstmParams.hpp:102
const TensorInfo * m_InputLayerNormWeights
Definition: LstmParams.hpp:106
const TensorInfo * m_CellToInputWeights
Definition: LstmParams.hpp:97
const TensorInfo * m_ProjectionWeights
Definition: LstmParams.hpp:104
const TensorInfo * m_ProjectionBias
Definition: LstmParams.hpp:105
const TensorInfo * m_InputToCellWeights
Definition: LstmParams.hpp:91
const TensorInfo * m_RecurrentToOutputWeights
Definition: LstmParams.hpp:96

Member Function Documentation

◆ Deref()

◆ GetCellBias()

const TensorInfo& GetCellBias ( ) const
inline

Definition at line 173 of file LstmParams.hpp.

174  {
175  return Deref(m_CellBias);
176  }
const TensorInfo & Deref(const TensorInfo *tensorInfo) const
Definition: LstmParams.hpp:111

References LstmInputParamsInfo::Deref(), and LstmInputParamsInfo::m_CellBias.

Referenced by RefLayerSupport::IsLstmSupported(), and RefLayerSupport::IsUnidirectionalSequenceLstmSupported().

◆ GetCellLayerNormWeights()

const TensorInfo& GetCellLayerNormWeights ( ) const
inline

◆ GetCellToForgetWeights()

const TensorInfo& GetCellToForgetWeights ( ) const
inline

◆ GetCellToInputWeights()

const TensorInfo& GetCellToInputWeights ( ) const
inline

◆ GetCellToOutputWeights()

const TensorInfo& GetCellToOutputWeights ( ) const
inline

◆ GetForgetGateBias()

const TensorInfo& GetForgetGateBias ( ) const
inline

◆ GetForgetLayerNormWeights()

const TensorInfo& GetForgetLayerNormWeights ( ) const
inline

◆ GetInputGateBias()

const TensorInfo& GetInputGateBias ( ) const
inline

◆ GetInputLayerNormWeights()

const TensorInfo& GetInputLayerNormWeights ( ) const
inline

◆ GetInputToCellWeights()

const TensorInfo& GetInputToCellWeights ( ) const
inline

◆ GetInputToForgetWeights()

const TensorInfo& GetInputToForgetWeights ( ) const
inline

◆ GetInputToInputWeights()

const TensorInfo& GetInputToInputWeights ( ) const
inline

◆ GetInputToOutputWeights()

const TensorInfo& GetInputToOutputWeights ( ) const
inline

◆ GetOutputGateBias()

const TensorInfo& GetOutputGateBias ( ) const
inline

◆ GetOutputLayerNormWeights()

const TensorInfo& GetOutputLayerNormWeights ( ) const
inline

◆ GetProjectionBias()

const TensorInfo& GetProjectionBias ( ) const
inline

◆ GetProjectionWeights()

const TensorInfo& GetProjectionWeights ( ) const
inline

◆ GetRecurrentToCellWeights()

const TensorInfo& GetRecurrentToCellWeights ( ) const
inline

◆ GetRecurrentToForgetWeights()

const TensorInfo& GetRecurrentToForgetWeights ( ) const
inline

◆ GetRecurrentToInputWeights()

const TensorInfo& GetRecurrentToInputWeights ( ) const
inline

◆ GetRecurrentToOutputWeights()

const TensorInfo& GetRecurrentToOutputWeights ( ) const
inline

Member Data Documentation

◆ m_CellBias

const TensorInfo* m_CellBias

Definition at line 102 of file LstmParams.hpp.

Referenced by LstmInputParamsInfo::GetCellBias().

◆ m_CellLayerNormWeights

const TensorInfo* m_CellLayerNormWeights

Definition at line 108 of file LstmParams.hpp.

Referenced by LstmInputParamsInfo::GetCellLayerNormWeights().

◆ m_CellToForgetWeights

const TensorInfo* m_CellToForgetWeights

Definition at line 98 of file LstmParams.hpp.

Referenced by LstmInputParamsInfo::GetCellToForgetWeights().

◆ m_CellToInputWeights

const TensorInfo* m_CellToInputWeights

Definition at line 97 of file LstmParams.hpp.

Referenced by LstmInputParamsInfo::GetCellToInputWeights().

◆ m_CellToOutputWeights

const TensorInfo* m_CellToOutputWeights

Definition at line 99 of file LstmParams.hpp.

Referenced by LstmInputParamsInfo::GetCellToOutputWeights().

◆ m_ForgetGateBias

const TensorInfo* m_ForgetGateBias

Definition at line 101 of file LstmParams.hpp.

Referenced by LstmInputParamsInfo::GetForgetGateBias().

◆ m_ForgetLayerNormWeights

const TensorInfo* m_ForgetLayerNormWeights

Definition at line 107 of file LstmParams.hpp.

Referenced by LstmInputParamsInfo::GetForgetLayerNormWeights().

◆ m_InputGateBias

const TensorInfo* m_InputGateBias

Definition at line 100 of file LstmParams.hpp.

Referenced by LstmInputParamsInfo::GetInputGateBias().

◆ m_InputLayerNormWeights

const TensorInfo* m_InputLayerNormWeights

Definition at line 106 of file LstmParams.hpp.

Referenced by LstmInputParamsInfo::GetInputLayerNormWeights().

◆ m_InputToCellWeights

const TensorInfo* m_InputToCellWeights

Definition at line 91 of file LstmParams.hpp.

Referenced by LstmInputParamsInfo::GetInputToCellWeights().

◆ m_InputToForgetWeights

const TensorInfo* m_InputToForgetWeights

Definition at line 90 of file LstmParams.hpp.

Referenced by LstmInputParamsInfo::GetInputToForgetWeights().

◆ m_InputToInputWeights

const TensorInfo* m_InputToInputWeights

Definition at line 89 of file LstmParams.hpp.

Referenced by LstmInputParamsInfo::GetInputToInputWeights().

◆ m_InputToOutputWeights

const TensorInfo* m_InputToOutputWeights

Definition at line 92 of file LstmParams.hpp.

Referenced by LstmInputParamsInfo::GetInputToOutputWeights().

◆ m_OutputGateBias

const TensorInfo* m_OutputGateBias

Definition at line 103 of file LstmParams.hpp.

Referenced by LstmInputParamsInfo::GetOutputGateBias().

◆ m_OutputLayerNormWeights

const TensorInfo* m_OutputLayerNormWeights

Definition at line 109 of file LstmParams.hpp.

Referenced by LstmInputParamsInfo::GetOutputLayerNormWeights().

◆ m_ProjectionBias

◆ m_ProjectionWeights

const TensorInfo* m_ProjectionWeights

Definition at line 104 of file LstmParams.hpp.

Referenced by LstmInputParamsInfo::GetProjectionWeights().

◆ m_RecurrentToCellWeights

const TensorInfo* m_RecurrentToCellWeights

Definition at line 95 of file LstmParams.hpp.

Referenced by LstmInputParamsInfo::GetRecurrentToCellWeights().

◆ m_RecurrentToForgetWeights

const TensorInfo* m_RecurrentToForgetWeights

Definition at line 94 of file LstmParams.hpp.

Referenced by LstmInputParamsInfo::GetRecurrentToForgetWeights().

◆ m_RecurrentToInputWeights

const TensorInfo* m_RecurrentToInputWeights

Definition at line 93 of file LstmParams.hpp.

Referenced by LstmInputParamsInfo::GetRecurrentToInputWeights().

◆ m_RecurrentToOutputWeights

const TensorInfo* m_RecurrentToOutputWeights

Definition at line 96 of file LstmParams.hpp.

Referenced by LstmInputParamsInfo::GetRecurrentToOutputWeights().


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