ArmNN
 21.02
RefPadWorkload Class Reference

#include <RefPadWorkload.hpp>

Inheritance diagram for RefPadWorkload:
BaseWorkload< PadQueueDescriptor > IWorkload

Public Member Functions

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

Detailed Description

Definition at line 14 of file RefPadWorkload.hpp.

Member Function Documentation

◆ Execute()

void Execute ( ) const
overridevirtual

Implements IWorkload.

Definition at line 15 of file RefPadWorkload.cpp.

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

16 {
17  ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefPadWorkload_Execute");
18 
19  const TensorInfo& inputInfo = GetTensorInfo(m_Data.m_Inputs[0]);
20  const TensorInfo& outputInfo = GetTensorInfo(m_Data.m_Outputs[0]);
21 
22  armnn::Pad(inputInfo,
23  outputInfo,
24  m_Data);
25 }
CPU Execution: Reference C++ kernels.
const PadQueueDescriptor m_Data
Definition: Workload.hpp:46
#define ARMNN_SCOPED_PROFILING_EVENT(backendId, name)
Definition: Profiling.hpp:173
std::vector< ITensorHandle * > m_Outputs
std::vector< ITensorHandle * > m_Inputs
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers

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