LstmOptPeepholeParameters Struct Reference

#include <LstmLayer.hpp>

Public Attributes
std::unique_ptr< ScopedCpuTensorHandle >	m_CellToInputWeights
	A unique pointer to represent 1D weights tensor with dimensions [num_units]. More...
std::unique_ptr< ScopedCpuTensorHandle >	m_CellToForgetWeights
	A unique pointer to represent 1D weights tensor with dimensions [num_units]. More...
std::unique_ptr< ScopedCpuTensorHandle >	m_CellToOutputWeights
	A unique pointer to represent 1D weights tensor with dimensions [num_units]. More...

Detailed Description

Definition at line 44 of file LstmLayer.hpp.

Member Data Documentation

m_CellToForgetWeights

std::unique_ptr<ScopedCpuTensorHandle> m_CellToForgetWeights

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

Definition at line 49 of file LstmLayer.hpp.

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

m_CellToInputWeights

std::unique_ptr<ScopedCpuTensorHandle> m_CellToInputWeights

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

Definition at line 47 of file LstmLayer.hpp.

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

m_CellToOutputWeights

std::unique_ptr<ScopedCpuTensorHandle> m_CellToOutputWeights

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

Definition at line 51 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