#include <MemCopyWorkload.hpp>

Inheritance diagram for CopyMemGenericWorkload:

Public Member Functions
	CopyMemGenericWorkload (const MemCopyQueueDescriptor &descriptor, const WorkloadInfo &info)

void	Execute () const override

void	ExecuteAsync (WorkingMemDescriptor &descriptor) override

Public Member Functions inherited from BaseWorkload< MemCopyQueueDescriptor >
	BaseWorkload (const MemCopyQueueDescriptor &descriptor, const WorkloadInfo &info)

void	ExecuteAsync (WorkingMemDescriptor &workingMemDescriptor) override

void	PostAllocationConfigure () override

const MemCopyQueueDescriptor &	GetData () const

arm::pipe::ProfilingGuid	GetGuid () const final

virtual bool	SupportsTensorHandleReplacement () const override

void	ReplaceInputTensorHandle (ITensorHandle *tensorHandle, unsigned int slot) override

void	ReplaceOutputTensorHandle (ITensorHandle *tensorHandle, unsigned int slot) override

Public Member Functions inherited from IWorkload
virtual	~IWorkload ()

virtual void	RegisterDebugCallback (const DebugCallbackFunction &)

Additional Inherited Members
Protected Attributes inherited from BaseWorkload< MemCopyQueueDescriptor >
MemCopyQueueDescriptor	m_Data

const arm::pipe::ProfilingGuid	m_Guid

Detailed Description

Definition at line 15 of file MemCopyWorkload.hpp.

Constructor & Destructor Documentation

◆ CopyMemGenericWorkload()

CopyMemGenericWorkload	(	const MemCopyQueueDescriptor &	descriptor,
		const WorkloadInfo &	info
	)

Definition at line 44 of file MemCopyWorkload.cpp.

References armnn::GatherTensorHandlePairs().

     : BaseWorkload<MemCopyQueueDescriptor>(descriptor, info)
 {
     GatherTensorHandlePairs(descriptor, m_TensorHandlePairs);
 }

Member Function Documentation

◆ Execute()

void Execute ( ) const

overridevirtual

Implements IWorkload.

Definition at line 51 of file MemCopyWorkload.cpp.

References ARMNN_SCOPED_PROFILING_EVENT, armnn::CopyTensorContentsGeneric(), and armnn::Undefined.

 {
     ARMNN_SCOPED_PROFILING_EVENT(Compute::Undefined, "CopyMemGeneric_Execute");
 
     auto copyFunc = [](void* dst, const void* src, size_t size)
         {
             memcpy(dst, src, size);
         };
 
     for (const auto& pair : m_TensorHandlePairs)
     {
         CopyTensorContentsGeneric(pair.first, pair.second, copyFunc);
     }
 }

◆ ExecuteAsync()

void ExecuteAsync ( WorkingMemDescriptor & descriptor )

overridevirtual

Implements IWorkload.

Definition at line 66 of file MemCopyWorkload.cpp.

References ARMNN_SCOPED_PROFILING_EVENT, armnn::CopyTensorContentsGeneric(), armnn::GatherTensorHandlePairs(), and armnn::Undefined.

 {
     ARMNN_SCOPED_PROFILING_EVENT(Compute::Undefined, "CopyMemGeneric_Execute_WorkingMemDescriptor");
     std::vector<TensorHandlePair> tensorHandlePairs;
     GatherTensorHandlePairs(descriptor, tensorHandlePairs);
 
     auto copyFunc = [](void* dst, const void* src, size_t size)
     {
         memcpy(dst, src, size);
     };
 
     for (const auto& pair : tensorHandlePairs)
     {
         CopyTensorContentsGeneric(pair.first, pair.second, copyFunc);
     }
 }

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

include/armnn/backends/MemCopyWorkload.hpp
src/backends/backendsCommon/MemCopyWorkload.cpp