ArmNN
 21.02
Public Member Functions | List of all members
RefFakeQuantizationFloat32Workload Class Reference

#include <RefFakeQuantizationFloat32Workload.hpp>

Inheritance diagram for RefFakeQuantizationFloat32Workload:
TypedWorkload< QueueDescriptor, DataTypes > BaseWorkload< QueueDescriptor > IWorkload

Public Member Functions

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

Detailed Description

Definition at line 14 of file RefFakeQuantizationFloat32Workload.hpp.

Member Function Documentation

◆ Execute()

void Execute ( ) const
overridevirtual

Implements IWorkload.

Definition at line 29 of file RefFakeQuantizationFloat32Workload.cpp.

References ARMNN_SCOPED_PROFILING_EVENT, armnn::CpuRef, armnn::FakeQuantization, armnn::GetInputTensorDataFloat(), TensorInfo::GetNumElements(), armnn::GetOutputTensorDataFloat(), armnn::GetTensorInfo(), BaseWorkload< QueueDescriptor >::m_Data, and QueueDescriptor::m_Inputs.

30 {
31  ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefFakeQuantizationFloat32Workload_Execute");
32 
33  const TensorInfo& inputInfo = GetTensorInfo(m_Data.m_Inputs[0]);
34 
35  const float* inputData = GetInputTensorDataFloat(0, m_Data);
36  float* outputData = GetOutputTensorDataFloat(0, m_Data);
37  FakeQuantization(inputData, outputData, inputInfo.GetNumElements(),
38  m_Data.m_Parameters.m_Min,
39  m_Data.m_Parameters.m_Max);
40 }
armnn::GetInputTensorDataFloat
const float * GetInputTensorDataFloat(unsigned int idx, const PayloadType &data)
Definition: RefWorkloadUtils.hpp:49
armnn::Compute::CpuRef
CPU Execution: Reference C++ kernels.
armnn::BaseWorkload::m_Data
const QueueDescriptor m_Data
Definition: Workload.hpp:46
ARMNN_SCOPED_PROFILING_EVENT
#define ARMNN_SCOPED_PROFILING_EVENT(backendId, name)
Definition: Profiling.hpp:173
armnn::QueueDescriptor::m_Inputs
std::vector< ITensorHandle * > m_Inputs
Definition: WorkloadData.hpp:30
armnn::GetOutputTensorDataFloat
float * GetOutputTensorDataFloat(unsigned int idx, const PayloadType &data)
Definition: RefWorkloadUtils.hpp:55
armnn::GetTensorInfo
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers
Definition: RefWorkloadUtils.hpp:26
The documentation for this class was generated from the following files: