ArmNN
 20.11
RefInstanceNormalizationWorkload Class Reference

#include <RefInstanceNormalizationWorkload.hpp>

Inheritance diagram for RefInstanceNormalizationWorkload:
BaseWorkload< InstanceNormalizationQueueDescriptor > IWorkload

Public Member Functions

 RefInstanceNormalizationWorkload (const InstanceNormalizationQueueDescriptor &descriptor, const WorkloadInfo &info)
 
virtual void Execute () const override
 
- Public Member Functions inherited from BaseWorkload< InstanceNormalizationQueueDescriptor >
 BaseWorkload (const InstanceNormalizationQueueDescriptor &descriptor, const WorkloadInfo &info)
 
void PostAllocationConfigure () override
 
const InstanceNormalizationQueueDescriptorGetData () 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< InstanceNormalizationQueueDescriptor >
const InstanceNormalizationQueueDescriptor m_Data
 
const profiling::ProfilingGuid m_Guid
 

Detailed Description

Definition at line 14 of file RefInstanceNormalizationWorkload.hpp.

Constructor & Destructor Documentation

◆ RefInstanceNormalizationWorkload()

RefInstanceNormalizationWorkload ( const InstanceNormalizationQueueDescriptor descriptor,
const WorkloadInfo info 
)
explicit

Definition at line 16 of file RefInstanceNormalizationWorkload.cpp.

19  : BaseWorkload<InstanceNormalizationQueueDescriptor>(descriptor, info) {}

Member Function Documentation

◆ Execute()

void Execute ( ) const
overridevirtual

Implements IWorkload.

Definition at line 21 of file RefInstanceNormalizationWorkload.cpp.

References ARMNN_SCOPED_PROFILING_EVENT, armnn::CpuRef, armnn::GetTensorInfo(), armnn::InstanceNorm(), BaseWorkload< InstanceNormalizationQueueDescriptor >::m_Data, QueueDescriptor::m_Inputs, and QueueDescriptor::m_Outputs.

22 {
23  ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefInstanceNormalizationWorkload_Execute");
24 
25  std::unique_ptr<Decoder<float>> inputDecoder = MakeDecoder<float>(GetTensorInfo(m_Data.m_Inputs[0]),
26  m_Data.m_Inputs[0]->Map());
27  std::unique_ptr<Encoder<float>> outputEncoder = MakeEncoder<float>(GetTensorInfo(m_Data.m_Outputs[0]),
28  m_Data.m_Outputs[0]->Map());
29 
30  InstanceNorm(m_Data, *inputDecoder, *outputEncoder);
31 }
CPU Execution: Reference C++ kernels.
const InstanceNormalizationQueueDescriptor m_Data
Definition: Workload.hpp:46
#define ARMNN_SCOPED_PROFILING_EVENT(backendId, name)
Definition: Profiling.hpp:175
std::vector< ITensorHandle * > m_Outputs
void InstanceNorm(const InstanceNormalizationQueueDescriptor &data, Decoder< float > &inputDecoder, Encoder< float > &outputEncoder)
std::vector< ITensorHandle * > m_Inputs
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers

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