ArmNN
 26.07
MultiTypedWorkload< QueueDescriptor, InputDataType, OutputDataType > Class Template Reference

#include <Workload.hpp>

Inheritance diagram for MultiTypedWorkload< QueueDescriptor, InputDataType, OutputDataType >:
[legend]
Collaboration diagram for MultiTypedWorkload< QueueDescriptor, InputDataType, OutputDataType >:
[legend]

Public Member Functions

 MultiTypedWorkload (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 PostAllocationConfigure () override
 
const QueueDescriptorGetData () 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 Execute () const =0
 
virtual void RegisterDebugCallback (const DebugCallbackFunction &)
 
virtual armnn::Optional< armnn::MemoryRequirementsGetMemoryRequirements ()
 

Additional Inherited Members

- Protected Attributes inherited from BaseWorkload< QueueDescriptor >
QueueDescriptor m_Data
 
const arm::pipe::ProfilingGuid m_Guid
 
const std::string m_Name
 

Detailed Description

template<typename QueueDescriptor, armnn::DataType InputDataType, armnn::DataType OutputDataType>
class armnn::MultiTypedWorkload< QueueDescriptor, InputDataType, OutputDataType >

Definition at line 140 of file Workload.hpp.

Constructor & Destructor Documentation

◆ MultiTypedWorkload()

MultiTypedWorkload ( const QueueDescriptor descriptor,
const WorkloadInfo info 
)
inline

Definition at line 144 of file Workload.hpp.

145  : BaseWorkload<QueueDescriptor>(descriptor, info)
146  {
147  if (std::all_of(info.m_InputTensorInfos.begin(),
148  info.m_InputTensorInfos.end(),
149  [&](auto it){
150  return it.GetDataType() == InputDataType;
151  }) == false)
152  {
153  throw armnn::Exception("Trying to create workload with incorrect type");
154  }
155  if (std::all_of(info.m_OutputTensorInfos.begin(),
156  info.m_OutputTensorInfos.end(),
157  [&](auto it){
158  return it.GetDataType() == OutputDataType;
159  }) == false)
160  {
161  throw armnn::Exception("Trying to create workload with incorrect type");
162  }
163  }
Base class for all ArmNN exceptions so that users can filter to just those.
Definition: Exceptions.hpp:47

References armnn::info.


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