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

#include <Workload.hpp>

Inheritance diagram for MultiTypedWorkload< QueueDescriptor, InputDataType, OutputDataType >:
BaseWorkload< QueueDescriptor > IWorkload ClConvertFp16ToFp32Workload ClConvertFp32ToFp16Workload NeonConvertFp16ToFp32Workload NeonConvertFp32ToFp16Workload RefConvertFp16ToFp32Workload RefConvertFp32ToFp16Workload

Public Member Functions

 MultiTypedWorkload (const QueueDescriptor &descriptor, const WorkloadInfo &info)
 
- Public Member Functions inherited from BaseWorkload< QueueDescriptor >
 BaseWorkload (const QueueDescriptor &descriptor, const WorkloadInfo &info)
 
void PostAllocationConfigure () override
 
const QueueDescriptorGetData () const
 
profiling::ProfilingGuid GetGuid () const final
 
- Public Member Functions inherited from IWorkload
virtual ~IWorkload ()
 
virtual void Execute () const =0
 
virtual void RegisterDebugCallback (const DebugCallbackFunction &)
 

Additional Inherited Members

- Protected Attributes inherited from BaseWorkload< QueueDescriptor >
const QueueDescriptor m_Data
 
const profiling::ProfilingGuid m_Guid
 

Detailed Description

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

Definition at line 105 of file Workload.hpp.

Constructor & Destructor Documentation

◆ MultiTypedWorkload()

MultiTypedWorkload ( const QueueDescriptor descriptor,
const WorkloadInfo info 
)
inline

Definition at line 109 of file Workload.hpp.

110  : BaseWorkload<QueueDescriptor>(descriptor, info)
111  {
112  BOOST_ASSERT_MSG(std::all_of(info.m_InputTensorInfos.begin(),
113  info.m_InputTensorInfos.end(),
114  [&](auto it){
115  return it.GetDataType() == InputDataType;
116  }),
117  "Trying to create workload with incorrect type");
118 
119  BOOST_ASSERT_MSG(std::all_of(info.m_OutputTensorInfos.begin(),
120  info.m_OutputTensorInfos.end(),
121  [&](auto it){
122  return it.GetDataType() == OutputDataType;
123  }),
124  "Trying to create workload with incorrect type");
125  }

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