21.11
|
Depthwise Convolution 2D layer workload data. More...
#include <WorkloadData.hpp>
Public Member Functions | |
DepthwiseConvolution2dQueueDescriptor () | |
void | Validate (const WorkloadInfo &workloadInfo) 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 |
![]() | |
DepthwiseConvolution2dDescriptor | m_Parameters |
![]() | |
std::vector< ITensorHandle * > | m_Inputs |
std::vector< ITensorHandle * > | m_Outputs |
void * | m_AdditionalInfoObject |
Additional Inherited Members | |
![]() | |
~QueueDescriptorWithParameters ()=default | |
QueueDescriptorWithParameters ()=default | |
QueueDescriptorWithParameters (QueueDescriptorWithParameters const &)=default | |
QueueDescriptorWithParameters & | operator= (QueueDescriptorWithParameters const &)=default |
![]() | |
~QueueDescriptor ()=default | |
QueueDescriptor () | |
QueueDescriptor (QueueDescriptor const &)=default | |
QueueDescriptor & | operator= (QueueDescriptor const &)=default |
Depthwise Convolution 2D layer workload data.
Definition at line 230 of file WorkloadData.hpp.
|
inline |
Definition at line 232 of file WorkloadData.hpp.
void Validate | ( | const WorkloadInfo & | workloadInfo | ) | const |
Definition at line 1381 of file WorkloadData.cpp.
References armnn::BFloat16, armnn::Float16, armnn::Float32, armnn::GetBiasDataType(), TensorInfo::GetDataType(), TensorInfo::GetShape(), WorkloadInfo::m_InputTensorInfos, WorkloadInfo::m_OutputTensorInfos, armnn::NCHW, armnn::QAsymmS8, armnn::QAsymmU8, armnn::QSymmS16, and OptionalReferenceSwitch< std::is_reference< T >::value, T >::value().
const ConstTensorHandle* m_Bias |
Definition at line 239 of file WorkloadData.hpp.
Referenced by ClDepthwiseConvolutionWorkload::ClDepthwiseConvolutionWorkload(), CompareDepthwiseConvolution2dTestImpl(), DepthwiseConvolution2dLayer::CreateWorkload(), DepthwiseConvolution2dDepthMul1TestImpl(), DepthwiseConvolution2dTestImpl(), and RefDepthwiseConvolution2dWorkload::RefDepthwiseConvolution2dWorkload().
const ConstTensorHandle* m_Weight |
Definition at line 238 of file WorkloadData.hpp.
Referenced by ClDepthwiseConvolutionWorkload::ClDepthwiseConvolutionWorkload(), CompareDepthwiseConvolution2dTestImpl(), DepthwiseConvolution2dLayer::CreateWorkload(), DepthwiseConvolution2dAsymmetricTestImpl(), DepthwiseConvolution2dDepthMul1TestImpl(), DepthwiseConvolution2dTestImpl(), NeonDepthwiseConvolutionWorkload::NeonDepthwiseConvolutionWorkload(), and RefDepthwiseConvolution2dWorkload::RefDepthwiseConvolution2dWorkload().