ArmNN
NotReleased
|
#include <WorkloadData.hpp>
Public Member Functions | |
LstmQueueDescriptor () | |
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 |
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 ()=default | |
QueueDescriptor (QueueDescriptor const &)=default | |
QueueDescriptor & | operator= (QueueDescriptor const &)=default |
Definition at line 358 of file WorkloadData.hpp.
|
inline |
Definition at line 360 of file WorkloadData.hpp.
void Validate | ( | const WorkloadInfo & | workloadInfo | ) | const |
Definition at line 1721 of file WorkloadData.cpp.
References armnn::Float16, armnn::Float32, WorkloadInfo::m_InputTensorInfos, WorkloadInfo::m_OutputTensorInfos, and armnn::QSymmS16.
const ConstCpuTensorHandle* m_CellBias |
Definition at line 398 of file WorkloadData.hpp.
Referenced by LstmLayer::CreateWorkload().
const ConstCpuTensorHandle* m_CellLayerNormWeights |
Definition at line 404 of file WorkloadData.hpp.
Referenced by LstmLayer::CreateWorkload().
const ConstCpuTensorHandle* m_CellToForgetWeights |
Definition at line 394 of file WorkloadData.hpp.
Referenced by LstmLayer::CreateWorkload().
const ConstCpuTensorHandle* m_CellToInputWeights |
Definition at line 393 of file WorkloadData.hpp.
Referenced by LstmLayer::CreateWorkload().
const ConstCpuTensorHandle* m_CellToOutputWeights |
Definition at line 395 of file WorkloadData.hpp.
Referenced by LstmLayer::CreateWorkload().
const ConstCpuTensorHandle* m_ForgetGateBias |
Definition at line 397 of file WorkloadData.hpp.
Referenced by LstmLayer::CreateWorkload().
const ConstCpuTensorHandle* m_ForgetLayerNormWeights |
Definition at line 403 of file WorkloadData.hpp.
Referenced by LstmLayer::CreateWorkload().
const ConstCpuTensorHandle* m_InputGateBias |
Definition at line 396 of file WorkloadData.hpp.
Referenced by LstmLayer::CreateWorkload().
const ConstCpuTensorHandle* m_InputLayerNormWeights |
Definition at line 402 of file WorkloadData.hpp.
Referenced by LstmLayer::CreateWorkload().
const ConstCpuTensorHandle* m_InputToCellWeights |
Definition at line 387 of file WorkloadData.hpp.
Referenced by LstmLayer::CreateWorkload().
const ConstCpuTensorHandle* m_InputToForgetWeights |
Definition at line 386 of file WorkloadData.hpp.
Referenced by LstmLayer::CreateWorkload().
const ConstCpuTensorHandle* m_InputToInputWeights |
Definition at line 385 of file WorkloadData.hpp.
Referenced by LstmLayer::CreateWorkload().
const ConstCpuTensorHandle* m_InputToOutputWeights |
Definition at line 388 of file WorkloadData.hpp.
Referenced by LstmLayer::CreateWorkload().
const ConstCpuTensorHandle* m_OutputGateBias |
Definition at line 399 of file WorkloadData.hpp.
Referenced by LstmLayer::CreateWorkload().
const ConstCpuTensorHandle* m_OutputLayerNormWeights |
Definition at line 405 of file WorkloadData.hpp.
Referenced by LstmLayer::CreateWorkload().
const ConstCpuTensorHandle* m_ProjectionBias |
Definition at line 401 of file WorkloadData.hpp.
Referenced by LstmLayer::CreateWorkload().
const ConstCpuTensorHandle* m_ProjectionWeights |
Definition at line 400 of file WorkloadData.hpp.
Referenced by LstmLayer::CreateWorkload().
const ConstCpuTensorHandle* m_RecurrentToCellWeights |
Definition at line 391 of file WorkloadData.hpp.
Referenced by LstmLayer::CreateWorkload().
const ConstCpuTensorHandle* m_RecurrentToForgetWeights |
Definition at line 390 of file WorkloadData.hpp.
Referenced by LstmLayer::CreateWorkload().
const ConstCpuTensorHandle* m_RecurrentToInputWeights |
Definition at line 389 of file WorkloadData.hpp.
Referenced by LstmLayer::CreateWorkload().
const ConstCpuTensorHandle* m_RecurrentToOutputWeights |
Definition at line 392 of file WorkloadData.hpp.
Referenced by LstmLayer::CreateWorkload().