ArmNN
 26.01
Loading...
Searching...
No Matches
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()

template<typename QueueDescriptor , armnn::DataType InputDataType, armnn::DataType OutputDataType>
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.

References armnn::info.


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