20.02
|
#include <WorkloadData.hpp>
Public Member Functions | |
BatchNormalizationQueueDescriptor () | |
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 |
Public Attributes | |
const ConstCpuTensorHandle * | m_Mean |
const ConstCpuTensorHandle * | m_Variance |
const ConstCpuTensorHandle * | m_Beta |
const ConstCpuTensorHandle * | m_Gamma |
Public Attributes inherited from QueueDescriptorWithParameters< BatchNormalizationDescriptor > | |
BatchNormalizationDescriptor | m_Parameters |
Public Attributes inherited from QueueDescriptor | |
std::vector< ITensorHandle * > | m_Inputs |
std::vector< ITensorHandle * > | m_Outputs |
Additional Inherited Members | |
Protected Member Functions inherited from QueueDescriptorWithParameters< BatchNormalizationDescriptor > | |
~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 269 of file WorkloadData.hpp.
|
inline |
Definition at line 271 of file WorkloadData.hpp.
void Validate | ( | const WorkloadInfo & | workloadInfo | ) | const |
Definition at line 1114 of file WorkloadData.cpp.
References armnn::BFloat16, armnn::Float16, armnn::Float32, WorkloadInfo::m_InputTensorInfos, WorkloadInfo::m_OutputTensorInfos, armnn::QAsymmU8, and armnn::QSymmS16.
const ConstCpuTensorHandle* m_Beta |
Definition at line 281 of file WorkloadData.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), CompareBatchNormTest(), and BatchNormalizationLayer::CreateWorkload().
const ConstCpuTensorHandle* m_Gamma |
Definition at line 282 of file WorkloadData.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), CompareBatchNormTest(), and BatchNormalizationLayer::CreateWorkload().
const ConstCpuTensorHandle* m_Mean |
Definition at line 279 of file WorkloadData.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), CompareBatchNormTest(), and BatchNormalizationLayer::CreateWorkload().
const ConstCpuTensorHandle* m_Variance |
Definition at line 280 of file WorkloadData.hpp.
Referenced by BOOST_AUTO_TEST_CASE(), CompareBatchNormTest(), and BatchNormalizationLayer::CreateWorkload().