ArmNN
 21.02
RefSliceWorkload Class Reference

#include <RefSliceWorkload.hpp>

Inheritance diagram for RefSliceWorkload:
BaseWorkload< SliceQueueDescriptor > IWorkload

Public Member Functions

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

Detailed Description

Definition at line 14 of file RefSliceWorkload.hpp.

Member Function Documentation

◆ Execute()

void Execute ( ) const
overridevirtual

Implements IWorkload.

Definition at line 16 of file RefSliceWorkload.cpp.

References ARMNN_SCOPED_PROFILING_EVENT, armnn::CpuRef, TensorInfo::GetDataType(), armnn::GetDataTypeSize(), armnn::GetTensorInfo(), BaseWorkload< SliceQueueDescriptor >::m_Data, QueueDescriptor::m_Inputs, QueueDescriptor::m_Outputs, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, and armnn::Slice.

17 {
18  ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefSliceWorkload_Execute");
19 
20  const TensorInfo& inputInfo = GetTensorInfo(m_Data.m_Inputs[0]);
21 
22  Slice(inputInfo,
24  m_Data.m_Inputs[0]->Map(),
25  m_Data.m_Outputs[0]->Map(),
26  GetDataTypeSize(inputInfo.GetDataType()));
27 }
CPU Execution: Reference C++ kernels.
const SliceQueueDescriptor 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
constexpr unsigned int GetDataTypeSize(DataType dataType)
Definition: TypesUtils.hpp:126

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