RefNormalizationWorkload Class Reference

#include <RefNormalizationWorkload.hpp>

Inheritance diagram for RefNormalizationWorkload:

Public Member Functions
	RefNormalizationWorkload (const NormalizationQueueDescriptor &descriptor, const WorkloadInfo &info)
virtual void	Execute () const override
Public Member Functions inherited from BaseWorkload< NormalizationQueueDescriptor >
	BaseWorkload (const NormalizationQueueDescriptor &descriptor, const WorkloadInfo &info)
void	PostAllocationConfigure () override
const NormalizationQueueDescriptor &	GetData () const
profiling::ProfilingGuid	GetGuid () const final
Public Member Functions inherited from IWorkload
virtual	~IWorkload ()
virtual void	RegisterDebugCallback (const DebugCallbackFunction &)

Additional Inherited Members
Protected Attributes inherited from BaseWorkload< NormalizationQueueDescriptor >
const NormalizationQueueDescriptor	m_Data
const profiling::ProfilingGuid	m_Guid

Detailed Description

Definition at line 14 of file RefNormalizationWorkload.hpp.

Constructor & Destructor Documentation

RefNormalizationWorkload()

RefNormalizationWorkload ( const NormalizationQueueDescriptor &  descriptor, const WorkloadInfo &  info  )

explicit

Definition at line 159 of file RefNormalizationWorkload.cpp.

    : BaseWorkload(descriptor, info)
{}

Member Function Documentation

Execute()

void Execute ( ) const

overridevirtual

Implements IWorkload.

Definition at line 164 of file RefNormalizationWorkload.cpp.

References armnn::Across, ARMNN_LOG, ARMNN_SCOPED_PROFILING_EVENT, armnn::CpuRef, armnn::GetTensorInfo(), armnn::LocalBrightness, NormalizationDescriptor::m_Alpha, NormalizationDescriptor::m_Beta, BaseWorkload< NormalizationQueueDescriptor >::m_Data, NormalizationDescriptor::m_DataLayout, QueueDescriptor::m_Inputs, NormalizationDescriptor::m_K, NormalizationDescriptor::m_NormChannelType, NormalizationDescriptor::m_NormMethodType, NormalizationDescriptor::m_NormSize, QueueDescriptor::m_Outputs, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, armnn::warning, and armnn::Within.

{
    ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefNormalizationWorkload_Execute");

    const TensorInfo& inputInfo = GetTensorInfo(m_Data.m_Inputs[0]);

    auto inputDecoder  = MakeDecoder<float>(inputInfo, m_Data.m_Inputs[0]->Map());
    auto outputEncoder = MakeEncoder<float>(inputInfo, m_Data.m_Outputs[0]->Map());

    if (NormalizationAlgorithmMethod::LocalBrightness == m_Data.m_Parameters.m_NormMethodType)
    {
        if (NormalizationAlgorithmChannel::Within == m_Data.m_Parameters.m_NormChannelType)
        {
            NormalizeWithinUingLbr(*inputDecoder,
                                   *outputEncoder,
                                   inputInfo.GetShape(),
                                   m_Data.m_Parameters.m_NormSize,
                                   m_Data.m_Parameters.m_Alpha,
                                   m_Data.m_Parameters.m_Beta,
                                   m_Data.m_Parameters.m_K);
        }
        else if (NormalizationAlgorithmChannel::Across == m_Data.m_Parameters.m_NormChannelType)
        {
            NormalizeAcrossUingLbr(*inputDecoder,
                                   *outputEncoder,
                                   inputInfo.GetShape(),
                                   m_Data.m_Parameters.m_NormSize,
                                   m_Data.m_Parameters.m_Alpha,
                                   m_Data.m_Parameters.m_Beta,
                                   m_Data.m_Parameters.m_K,
                                   m_Data.m_Parameters.m_DataLayout);
        }
        else
        {
            ARMNN_LOG(warning) << "Illegal NORMALIZATION mode in normalization_f32";
            return;
        }
    }
    else
    {
        ARMNN_LOG(warning) << "Lcr method (Jarret 2009: Local Contrast Normalization) not supported yet.";
        return;
    }
}

---

The documentation for this class was generated from the following files:

• src/backends/reference/workloads/RefNormalizationWorkload.hpp
• src/backends/reference/workloads/RefNormalizationWorkload.cpp