24.08
|
#include <WorkloadData.hpp>
Public Member Functions | |
QuantizedLstmQueueDescriptor () | |
void | Validate (const WorkloadInfo &workloadInfo) const |
Public Member Functions inherited from QueueDescriptor | |
virtual | ~QueueDescriptor ()=default |
void | ValidateTensorNumDimensions (const TensorInfo &tensor, std::string const &descName, unsigned int numDimensions, std::string const &tensorName) const |
void | ValidateTensorNumDimNumElem (const TensorInfo &tensorInfo, unsigned int numDimension, unsigned int numElements, std::string const &tensorName) const |
void | ValidateInputsOutputs (const std::string &descName, unsigned int numExpectedIn, unsigned int numExpectedOut) const |
template<typename T > | |
const T * | GetAdditionalInformation () const |
Public Attributes | |
const ConstTensorHandle * | m_InputToInputWeights |
const ConstTensorHandle * | m_InputToForgetWeights |
const ConstTensorHandle * | m_InputToCellWeights |
const ConstTensorHandle * | m_InputToOutputWeights |
const ConstTensorHandle * | m_RecurrentToInputWeights |
const ConstTensorHandle * | m_RecurrentToForgetWeights |
const ConstTensorHandle * | m_RecurrentToCellWeights |
const ConstTensorHandle * | m_RecurrentToOutputWeights |
const ConstTensorHandle * | m_InputGateBias |
const ConstTensorHandle * | m_ForgetGateBias |
const ConstTensorHandle * | m_CellBias |
const ConstTensorHandle * | m_OutputGateBias |
Public Attributes inherited from QueueDescriptor | |
std::vector< ITensorHandle * > | m_Inputs |
std::vector< ITensorHandle * > | m_Outputs |
void * | m_AdditionalInfoObject |
bool | m_AllowExpandedDims = false |
Additional Inherited Members | |
Protected Member Functions inherited from QueueDescriptor | |
QueueDescriptor () | |
QueueDescriptor (QueueDescriptor const &)=default | |
QueueDescriptor & | operator= (QueueDescriptor const &)=default |
Definition at line 614 of file WorkloadData.hpp.
|
inline |
Definition at line 616 of file WorkloadData.hpp.
void Validate | ( | const WorkloadInfo & | workloadInfo | ) | const |
Definition at line 3436 of file WorkloadData.cpp.
References TensorInfo::GetShape(), ConstTensorHandle::GetTensorInfo(), QuantizedLstmQueueDescriptor::m_CellBias, QuantizedLstmQueueDescriptor::m_ForgetGateBias, QuantizedLstmQueueDescriptor::m_InputGateBias, WorkloadInfo::m_InputTensorInfos, QuantizedLstmQueueDescriptor::m_InputToCellWeights, QuantizedLstmQueueDescriptor::m_InputToForgetWeights, QuantizedLstmQueueDescriptor::m_InputToInputWeights, QuantizedLstmQueueDescriptor::m_InputToOutputWeights, QuantizedLstmQueueDescriptor::m_OutputGateBias, WorkloadInfo::m_OutputTensorInfos, QuantizedLstmQueueDescriptor::m_RecurrentToCellWeights, QuantizedLstmQueueDescriptor::m_RecurrentToForgetWeights, QuantizedLstmQueueDescriptor::m_RecurrentToInputWeights, QuantizedLstmQueueDescriptor::m_RecurrentToOutputWeights, armnn::QAsymmU8, armnn::QSymmS16, armnn::Signed32, and QueueDescriptor::ValidateTensorNumDimNumElem().
const ConstTensorHandle* m_CellBias |
Definition at line 645 of file WorkloadData.hpp.
Referenced by QuantizedLstmLayer::CreateWorkload(), and QuantizedLstmQueueDescriptor::Validate().
const ConstTensorHandle* m_ForgetGateBias |
Definition at line 644 of file WorkloadData.hpp.
Referenced by QuantizedLstmLayer::CreateWorkload(), and QuantizedLstmQueueDescriptor::Validate().
const ConstTensorHandle* m_InputGateBias |
Definition at line 643 of file WorkloadData.hpp.
Referenced by QuantizedLstmLayer::CreateWorkload(), and QuantizedLstmQueueDescriptor::Validate().
const ConstTensorHandle* m_InputToCellWeights |
Definition at line 635 of file WorkloadData.hpp.
Referenced by QuantizedLstmLayer::CreateWorkload(), and QuantizedLstmQueueDescriptor::Validate().
const ConstTensorHandle* m_InputToForgetWeights |
Definition at line 634 of file WorkloadData.hpp.
Referenced by QuantizedLstmLayer::CreateWorkload(), and QuantizedLstmQueueDescriptor::Validate().
const ConstTensorHandle* m_InputToInputWeights |
Definition at line 633 of file WorkloadData.hpp.
Referenced by QuantizedLstmLayer::CreateWorkload(), and QuantizedLstmQueueDescriptor::Validate().
const ConstTensorHandle* m_InputToOutputWeights |
Definition at line 636 of file WorkloadData.hpp.
Referenced by QuantizedLstmLayer::CreateWorkload(), and QuantizedLstmQueueDescriptor::Validate().
const ConstTensorHandle* m_OutputGateBias |
Definition at line 646 of file WorkloadData.hpp.
Referenced by QuantizedLstmLayer::CreateWorkload(), and QuantizedLstmQueueDescriptor::Validate().
const ConstTensorHandle* m_RecurrentToCellWeights |
Definition at line 640 of file WorkloadData.hpp.
Referenced by QuantizedLstmLayer::CreateWorkload(), and QuantizedLstmQueueDescriptor::Validate().
const ConstTensorHandle* m_RecurrentToForgetWeights |
Definition at line 639 of file WorkloadData.hpp.
Referenced by QuantizedLstmLayer::CreateWorkload(), and QuantizedLstmQueueDescriptor::Validate().
const ConstTensorHandle* m_RecurrentToInputWeights |
Definition at line 638 of file WorkloadData.hpp.
Referenced by QuantizedLstmLayer::CreateWorkload(), and QuantizedLstmQueueDescriptor::Validate().
const ConstTensorHandle* m_RecurrentToOutputWeights |
Definition at line 641 of file WorkloadData.hpp.
Referenced by QuantizedLstmLayer::CreateWorkload(), and QuantizedLstmQueueDescriptor::Validate().