LstmOptCifgParameters Struct Reference

#include <LstmLayer.hpp>

Public Attributes
std::unique_ptr< ScopedCpuTensorHandle >	m_InputToInputWeights
	A unique pointer to represent 2D weights tensor with dimensions [input_size, num_units]. More...
std::unique_ptr< ScopedCpuTensorHandle >	m_RecurrentToInputWeights
	A unique pointer to represent 2D weights tensor with dimensions [input_size, num_units]. More...
std::unique_ptr< ScopedCpuTensorHandle >	m_InputGateBias
	A unique pointer to represent 1D weights tensor with dimensions [num_units]. More...

Detailed Description

Definition at line 26 of file LstmLayer.hpp.

Member Data Documentation

m_InputGateBias

std::unique_ptr<ScopedCpuTensorHandle> m_InputGateBias

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

Definition at line 33 of file LstmLayer.hpp.

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

m_InputToInputWeights

std::unique_ptr<ScopedCpuTensorHandle> m_InputToInputWeights

A unique pointer to represent 2D weights tensor with dimensions [input_size, num_units].

Definition at line 29 of file LstmLayer.hpp.

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

m_RecurrentToInputWeights

std::unique_ptr<ScopedCpuTensorHandle> m_RecurrentToInputWeights

A unique pointer to represent 2D weights tensor with dimensions [input_size, num_units].

Definition at line 31 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:

• src/armnn/layers/LstmLayer.hpp