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

#include <LstmLayer.hpp>

Public Attributes

std::unique_ptr< ScopedCpuTensorHandle > m_InputLayerNormWeights
 A unique pointer to represent 1D weights tensor with dimensions [num_units]. More...
 
std::unique_ptr< ScopedCpuTensorHandle > m_ForgetLayerNormWeights
 A unique pointer to represent 1D weights tensor with dimensions [num_units]. More...
 
std::unique_ptr< ScopedCpuTensorHandle > m_CellLayerNormWeights
 A unique pointer to represent 1D weights tensor with dimensions [num_units]. More...
 
std::unique_ptr< ScopedCpuTensorHandle > m_OutputLayerNormWeights
 A unique pointer to represent 1D weights tensor with dimensions [num_units]. More...
 

Detailed Description

Definition at line 14 of file LstmLayer.hpp.

Member Data Documentation

◆ m_CellLayerNormWeights

std::unique_ptr<ScopedCpuTensorHandle> m_CellLayerNormWeights

A unique pointer to represent 1D weights tensor with dimensions [num_units].

Definition at line 21 of file LstmLayer.hpp.

Referenced by LstmLayer::Accept(), LstmLayer::Clone(), LstmLayer::CreateWorkload(), LstmLayer::ExecuteStrategy(), and LstmLayer::GetConstantTensorsByRef().

◆ m_ForgetLayerNormWeights

std::unique_ptr<ScopedCpuTensorHandle> m_ForgetLayerNormWeights

A unique pointer to represent 1D weights tensor with dimensions [num_units].

Definition at line 19 of file LstmLayer.hpp.

Referenced by LstmLayer::Accept(), LstmLayer::Clone(), LstmLayer::CreateWorkload(), LstmLayer::ExecuteStrategy(), and LstmLayer::GetConstantTensorsByRef().

◆ m_InputLayerNormWeights

std::unique_ptr<ScopedCpuTensorHandle> m_InputLayerNormWeights

A unique pointer to represent 1D weights tensor with dimensions [num_units].

Definition at line 17 of file LstmLayer.hpp.

Referenced by LstmLayer::Accept(), LstmLayer::Clone(), LstmLayer::CreateWorkload(), LstmLayer::ExecuteStrategy(), and LstmLayer::GetConstantTensorsByRef().

◆ m_OutputLayerNormWeights

std::unique_ptr<ScopedCpuTensorHandle> m_OutputLayerNormWeights

A unique pointer to represent 1D weights tensor with dimensions [num_units].

Definition at line 23 of file LstmLayer.hpp.

Referenced by LstmLayer::Accept(), LstmLayer::Clone(), LstmLayer::CreateWorkload(), LstmLayer::ExecuteStrategy(), and LstmLayer::GetConstantTensorsByRef().

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