|
25.02
|
#include <WorkloadData.hpp>
Public Member Functions | |
| TransposeConvolution2dQueueDescriptor () | |
| void | Validate (const WorkloadInfo &workloadInfo) const |
Public Member Functions inherited from QueueDescriptorWithParameters< TransposeConvolution2dDescriptor > | |
| virtual | ~QueueDescriptorWithParameters ()=default |
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_Weight |
| const ConstTensorHandle * | m_Bias |
Public Attributes inherited from QueueDescriptorWithParameters< TransposeConvolution2dDescriptor > | |
| TransposeConvolution2dDescriptor | m_Parameters |
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 QueueDescriptorWithParameters< TransposeConvolution2dDescriptor > | |
| QueueDescriptorWithParameters ()=default | |
| QueueDescriptorWithParameters (QueueDescriptorWithParameters const &)=default | |
| QueueDescriptorWithParameters & | operator= (QueueDescriptorWithParameters const &)=default |
Protected Member Functions inherited from QueueDescriptor | |
| QueueDescriptor () | |
| QueueDescriptor (QueueDescriptor const &)=default | |
| QueueDescriptor & | operator= (QueueDescriptor const &)=default |
Definition at line 544 of file WorkloadData.hpp.
|
inline |
Definition at line 546 of file WorkloadData.hpp.
| void Validate | ( | const WorkloadInfo & | workloadInfo | ) | const |
Definition at line 3066 of file WorkloadData.cpp.
References armnn::BFloat16, armnn::Float16, armnn::Float32, armnn::GetBiasDataType(), TensorInfo::GetDataType(), ConstTensorHandle::GetTensorInfo(), TransposeConvolution2dQueueDescriptor::m_Bias, TransposeConvolution2dDescriptor::m_BiasEnabled, WorkloadInfo::m_InputTensorInfos, WorkloadInfo::m_OutputTensorInfos, QueueDescriptorWithParameters< TransposeConvolution2dDescriptor >::m_Parameters, TransposeConvolution2dQueueDescriptor::m_Weight, armnn::QAsymmS8, armnn::QAsymmU8, armnn::QSymmS16, QueueDescriptor::ValidateTensorNumDimensions(), and OptionalReferenceSwitch< IsReference, T >::value().
| const ConstTensorHandle* m_Bias |
Definition at line 552 of file WorkloadData.hpp.
Referenced by ClTransposeConvolution2dWorkload::ClTransposeConvolution2dWorkload(), TransposeConvolution2dLayer::CreateWorkload(), RefTransposeConvolution2dWorkload::RefTransposeConvolution2dWorkload(), and TransposeConvolution2dQueueDescriptor::Validate().
| const ConstTensorHandle* m_Weight |
Definition at line 551 of file WorkloadData.hpp.
Referenced by ClTransposeConvolution2dWorkload::ClTransposeConvolution2dWorkload(), TransposeConvolution2dLayer::CreateWorkload(), RefTransposeConvolution2dWorkload::RefTransposeConvolution2dWorkload(), and TransposeConvolution2dQueueDescriptor::Validate().