ArmNN
 20.05
RefDequantizeWorkload Class Reference

#include <RefDequantizeWorkload.hpp>

Inheritance diagram for RefDequantizeWorkload:
BaseWorkload< DequantizeQueueDescriptor > IWorkload

Public Member Functions

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

Detailed Description

Definition at line 13 of file RefDequantizeWorkload.hpp.

Member Function Documentation

◆ Execute()

void Execute ( ) const
overridevirtual

Implements IWorkload.

Definition at line 15 of file RefDequantizeWorkload.cpp.

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

16 {
17  ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefDequantizeWorkload_Execute");
18 
19  const TensorInfo& inputInfo = GetTensorInfo(m_Data.m_Inputs[0]);
20  const TensorInfo& outputInfo = GetTensorInfo(m_Data.m_Outputs[0]);
21 
22  auto inputDecoder = MakeDecoder<float>(inputInfo, m_Data.m_Inputs[0]->Map());
23  auto outputEncoder = MakeEncoder<float>(outputInfo, m_Data.m_Outputs[0]->Map());
24 
25  Dequantize(*inputDecoder, *outputEncoder, inputInfo, outputInfo);
26 }
CPU Execution: Reference C++ kernels.
const DequantizeQueueDescriptor m_Data
Definition: Workload.hpp:46
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers
#define ARMNN_SCOPED_PROFILING_EVENT(backendId, name)
Definition: Profiling.hpp:169
std::vector< ITensorHandle * > m_Outputs
std::vector< ITensorHandle * > m_Inputs

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