21.08
|
#include <WorkloadData.hpp>
Public Member Functions | |
UnidirectionalSequenceLstmQueueDescriptor () | |
void | Validate (const WorkloadInfo &workloadInfo) const |
Public Member Functions inherited from QueueDescriptor | |
void | ValidateInputsOutputs (const std::string &descName, unsigned int numExpectedIn, unsigned int numExpectedOut) const |
template<typename T > | |
const T * | GetAdditionalInformation () const |
Additional Inherited Members | |
Protected Member Functions inherited from QueueDescriptorWithParameters< LstmDescriptor > | |
~QueueDescriptorWithParameters ()=default | |
QueueDescriptorWithParameters ()=default | |
QueueDescriptorWithParameters (QueueDescriptorWithParameters const &)=default | |
QueueDescriptorWithParameters & | operator= (QueueDescriptorWithParameters const &)=default |
Protected Member Functions inherited from QueueDescriptor | |
~QueueDescriptor ()=default | |
QueueDescriptor () | |
QueueDescriptor (QueueDescriptor const &)=default | |
QueueDescriptor & | operator= (QueueDescriptor const &)=default |
Definition at line 698 of file WorkloadData.hpp.
|
inline |
Definition at line 700 of file WorkloadData.hpp.
void Validate | ( | const WorkloadInfo & | workloadInfo | ) | const |
Definition at line 3719 of file WorkloadData.cpp.
References armnn::Float32, WorkloadInfo::m_InputTensorInfos, and WorkloadInfo::m_OutputTensorInfos.
const ConstTensorHandle* m_CellBias |
Definition at line 738 of file WorkloadData.hpp.
Referenced by UnidirectionalSequenceLstmLayer::CreateWorkload(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), and UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest().
const ConstTensorHandle* m_CellLayerNormWeights |
Definition at line 744 of file WorkloadData.hpp.
Referenced by UnidirectionalSequenceLstmLayer::CreateWorkload(), and UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest().
const ConstTensorHandle* m_CellToForgetWeights |
Definition at line 734 of file WorkloadData.hpp.
Referenced by UnidirectionalSequenceLstmLayer::CreateWorkload(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), and UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest().
const ConstTensorHandle* m_CellToInputWeights |
const ConstTensorHandle* m_CellToOutputWeights |
Definition at line 735 of file WorkloadData.hpp.
Referenced by UnidirectionalSequenceLstmLayer::CreateWorkload(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), and UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest().
const ConstTensorHandle* m_ForgetGateBias |
Definition at line 737 of file WorkloadData.hpp.
Referenced by UnidirectionalSequenceLstmLayer::CreateWorkload(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), and UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest().
const ConstTensorHandle* m_ForgetLayerNormWeights |
Definition at line 743 of file WorkloadData.hpp.
Referenced by UnidirectionalSequenceLstmLayer::CreateWorkload(), and UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest().
const ConstTensorHandle* m_InputGateBias |
const ConstTensorHandle* m_InputLayerNormWeights |
Definition at line 742 of file WorkloadData.hpp.
Referenced by UnidirectionalSequenceLstmLayer::CreateWorkload(), and UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest().
const ConstTensorHandle* m_InputToCellWeights |
Definition at line 727 of file WorkloadData.hpp.
Referenced by UnidirectionalSequenceLstmLayer::CreateWorkload(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), and UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest().
const ConstTensorHandle* m_InputToForgetWeights |
Definition at line 726 of file WorkloadData.hpp.
Referenced by UnidirectionalSequenceLstmLayer::CreateWorkload(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), and UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest().
const ConstTensorHandle* m_InputToInputWeights |
const ConstTensorHandle* m_InputToOutputWeights |
Definition at line 728 of file WorkloadData.hpp.
Referenced by UnidirectionalSequenceLstmLayer::CreateWorkload(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), and UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest().
const ConstTensorHandle* m_OutputGateBias |
Definition at line 739 of file WorkloadData.hpp.
Referenced by UnidirectionalSequenceLstmLayer::CreateWorkload(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), and UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest().
const ConstTensorHandle* m_OutputLayerNormWeights |
Definition at line 745 of file WorkloadData.hpp.
Referenced by UnidirectionalSequenceLstmLayer::CreateWorkload(), and UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest().
const ConstTensorHandle* m_ProjectionBias |
const ConstTensorHandle* m_ProjectionWeights |
const ConstTensorHandle* m_RecurrentToCellWeights |
Definition at line 731 of file WorkloadData.hpp.
Referenced by UnidirectionalSequenceLstmLayer::CreateWorkload(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), and UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest().
const ConstTensorHandle* m_RecurrentToForgetWeights |
Definition at line 730 of file WorkloadData.hpp.
Referenced by UnidirectionalSequenceLstmLayer::CreateWorkload(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), and UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest().
const ConstTensorHandle* m_RecurrentToInputWeights |
const ConstTensorHandle* m_RecurrentToOutputWeights |
Definition at line 732 of file WorkloadData.hpp.
Referenced by UnidirectionalSequenceLstmLayer::CreateWorkload(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionTest(), UnidirectionalSequenceLstmLayerNoCifgWithPeepholeWithProjectionWithLayerNormTest(), and UnidirectionalSequenceLstmWithCifgWithPeepholeNoProjectionTest().