24.08
|
#include <ClBatchNormalizationFloatWorkload.hpp>
Public Member Functions | |
ClBatchNormalizationFloatWorkload (const BatchNormalizationQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext) | |
void | Execute () const override |
void | ReplaceInputTensorHandle (ITensorHandle *tensorHandle, unsigned int slot) override |
void | ReplaceOutputTensorHandle (ITensorHandle *tensorHandle, unsigned int slot) 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) | |
virtual const std::string & | GetName () const override |
void | ExecuteAsync (ExecutionData &executionData) override |
void | PostAllocationConfigure () override |
const QueueDescriptor & | GetData () const |
arm::pipe::ProfilingGuid | GetGuid () const final |
virtual bool | SupportsTensorHandleReplacement () const override |
Public Member Functions inherited from IWorkload | |
virtual | ~IWorkload () |
virtual void | RegisterDebugCallback (const DebugCallbackFunction &) |
virtual armnn::Optional< armnn::MemoryRequirements > | GetMemoryRequirements () |
Additional Inherited Members | |
Protected Attributes inherited from BaseWorkload< QueueDescriptor > | |
QueueDescriptor | m_Data |
const arm::pipe::ProfilingGuid | m_Guid |
const std::string | m_Name |
Definition at line 25 of file ClBatchNormalizationFloatWorkload.hpp.
ClBatchNormalizationFloatWorkload | ( | const BatchNormalizationQueueDescriptor & | descriptor, |
const WorkloadInfo & | info, | ||
const arm_compute::CLCompileContext & | clCompileContext | ||
) |
Definition at line 54 of file ClBatchNormalizationFloatWorkload.cpp.
References ARMNN_REPORT_PROFILING_WORKLOAD_DESC, armnn::info, and QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters.
|
overridevirtual |
Implements IWorkload.
Definition at line 113 of file ClBatchNormalizationFloatWorkload.cpp.
References ARMNN_SCOPED_PROFILING_EVENT_CL_NAME_GUID, CHECK_LOCATION, and armnn::RunClFunction().
|
overridevirtual |
Reimplemented from BaseWorkload< QueueDescriptor >.
Definition at line 127 of file ClBatchNormalizationFloatWorkload.cpp.
References BaseWorkload< QueueDescriptor >::m_Data, and QueueDescriptor::m_Inputs.
|
overridevirtual |
Reimplemented from BaseWorkload< QueueDescriptor >.
Definition at line 144 of file ClBatchNormalizationFloatWorkload.cpp.
References BaseWorkload< QueueDescriptor >::m_Data, and QueueDescriptor::m_Inputs.