From 6940dd720ebb6b3d1df8ca203ab696daefe58189 Mon Sep 17 00:00:00 2001 From: Jim Flynn Date: Fri, 20 Mar 2020 12:25:56 +0000 Subject: renamed Documentation folder 20.02 and added .nojekyll file Signed-off-by: Jim Flynn --- 20.02/classarmnn_1_1_ref_workload_factory.xhtml | 3309 +++++++++++++++++++++++ 1 file changed, 3309 insertions(+) create mode 100644 20.02/classarmnn_1_1_ref_workload_factory.xhtml (limited to '20.02/classarmnn_1_1_ref_workload_factory.xhtml') diff --git a/20.02/classarmnn_1_1_ref_workload_factory.xhtml b/20.02/classarmnn_1_1_ref_workload_factory.xhtml new file mode 100644 index 0000000000..9cab100bc6 --- /dev/null +++ b/20.02/classarmnn_1_1_ref_workload_factory.xhtml @@ -0,0 +1,3309 @@ + + + + + + + + + + + + + +ArmNN: RefWorkloadFactory Class Reference + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  20.02 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+Public Member Functions | +Static Public Member Functions | +List of all members
+
+
RefWorkloadFactory Class Reference
+
+
+ +

#include <RefWorkloadFactory.hpp>

+
+Inheritance diagram for RefWorkloadFactory:
+
+
+ + +IWorkloadFactory + +
+ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +

+Public Member Functions

 RefWorkloadFactory (const std::shared_ptr< RefMemoryManager > &memoryManager)
 
 RefWorkloadFactory ()
 
 ~RefWorkloadFactory ()
 
const BackendId & GetBackendId () const override
 
bool SupportsSubTensors () const override
 
std::unique_ptr< ITensorHandle > CreateSubTensorHandle (ITensorHandle &parent, TensorShape const &subTensorShape, unsigned int const *subTensorOrigin) const override
 
std::unique_ptr< ITensorHandle > CreateTensorHandle (const TensorInfo &tensorInfo, const bool IsMemoryManaged=true) const override
 
std::unique_ptr< ITensorHandle > CreateTensorHandle (const TensorInfo &tensorInfo, DataLayout dataLayout, const bool IsMemoryManaged=true) const override
 
std::unique_ptr< IWorkload > CreateAbs (const AbsQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateActivation (const ActivationQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateAddition (const AdditionQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateArgMinMax (const ArgMinMaxQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateBatchNormalization (const BatchNormalizationQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateBatchToSpaceNd (const BatchToSpaceNdQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateComparison (const ComparisonQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateConcat (const ConcatQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateConstant (const ConstantQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateConvertFp16ToFp32 (const ConvertFp16ToFp32QueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateConvertFp32ToFp16 (const ConvertFp32ToFp16QueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateConvolution2d (const Convolution2dQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateDebug (const DebugQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateDepthToSpace (const DepthToSpaceQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateDepthwiseConvolution2d (const DepthwiseConvolution2dQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateDequantize (const DequantizeQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateDetectionPostProcess (const DetectionPostProcessQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateDivision (const DivisionQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateElementwiseUnary (const ElementwiseUnaryQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateEqual (const EqualQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateFakeQuantization (const FakeQuantizationQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateFloor (const FloorQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateFullyConnected (const FullyConnectedQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateGather (const GatherQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateGreater (const GreaterQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateInput (const InputQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateInstanceNormalization (const InstanceNormalizationQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateL2Normalization (const L2NormalizationQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateLogSoftmax (const LogSoftmaxQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateLstm (const LstmQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateMaximum (const MaximumQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateMean (const MeanQueueDescriptor &descriptor, const WorkloadInfo &Info) const override
 
std::unique_ptr< IWorkload > CreateMemCopy (const MemCopyQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateMemImport (const MemImportQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateMerger (const MergerQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateMinimum (const MinimumQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateMultiplication (const MultiplicationQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateNormalization (const NormalizationQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateOutput (const OutputQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreatePad (const PadQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreatePermute (const PermuteQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreatePooling2d (const Pooling2dQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreatePreCompiled (const PreCompiledQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreatePrelu (const PreluQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateQuantize (const QuantizeQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateReshape (const ReshapeQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateResize (const ResizeQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateResizeBilinear (const ResizeBilinearQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateRsqrt (const RsqrtQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateSlice (const SliceQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateSoftmax (const SoftmaxQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateSpaceToBatchNd (const SpaceToBatchNdQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateSpaceToDepth (const SpaceToDepthQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateSplitter (const SplitterQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateStack (const StackQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateStridedSlice (const StridedSliceQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateSubtraction (const SubtractionQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateTranspose (const TransposeQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
std::unique_ptr< IWorkload > CreateTransposeConvolution2d (const TransposeConvolution2dQueueDescriptor &descriptor, const WorkloadInfo &info) const override
 
- Public Member Functions inherited from IWorkloadFactory
virtual ~IWorkloadFactory ()
 
virtual std::unique_ptr< IWorkload > CreateMerge (const MergeQueueDescriptor &descriptor, const WorkloadInfo &info) const
 
virtual std::unique_ptr< IWorkload > CreateQuantizedLstm (const QuantizedLstmQueueDescriptor &descriptor, const WorkloadInfo &info) const
 
virtual std::unique_ptr< IWorkload > CreateSwitch (const SwitchQueueDescriptor &descriptor, const WorkloadInfo &Info) const
 
+ + + + + + + + +

+Static Public Member Functions

static bool IsLayerSupported (const Layer &layer, Optional< DataType > dataType, std::string &outReasonIfUnsupported)
 
- Static Public Member Functions inherited from IWorkloadFactory
static bool IsLayerSupported (const BackendId &backendId, const IConnectableLayer &layer, Optional< DataType > dataType, std::string &outReasonIfUnsupported)
 
static bool IsLayerSupported (const IConnectableLayer &layer, Optional< DataType > dataType, std::string &outReasonIfUnsupported)
 
+

Detailed Description

+
+

Definition at line 30 of file RefWorkloadFactory.hpp.

+

Constructor & Destructor Documentation

+ +

◆ RefWorkloadFactory() [1/2]

+ +
+
+ + + + + +
+ + + + + + + + +
RefWorkloadFactory (const std::shared_ptr< RefMemoryManager > & memoryManager)
+ 		+explicit
+
+ +

Definition at line 83 of file RefWorkloadFactory.cpp.

+
84  : m_MemoryManager(memoryManager)
85 {
86 }
+
+
+ +

◆ RefWorkloadFactory() [2/2]

+ +
+
+ + + + + + + +
RefWorkloadFactory ()
+
+ +

Definition at line 88 of file RefWorkloadFactory.cpp.

+
89  : m_MemoryManager(new RefMemoryManager())
90 {
91 }
+
+
+ +

◆ ~RefWorkloadFactory()

+ +
+
+ + + + + +
+ + + + + + + +
~RefWorkloadFactory ()
+ 		+inline
+
+ +

Definition at line 36 of file RefWorkloadFactory.hpp.

+ +

References RefWorkloadFactory::GetBackendId(), and RefWorkloadFactory::IsLayerSupported().

+
36 {}
+
+
+

Member Function Documentation

+ +

◆ CreateAbs()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateAbs (const AbsQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 124 of file RefWorkloadFactory.cpp.

+ +

References armnn::Abs, RefWorkloadFactory::CreateElementwiseUnary(), armnn::IgnoreUnused(), ElementwiseUnaryDescriptor::m_Operation, and QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
126 {
127  IgnoreUnused(descriptor);
128  ElementwiseUnaryQueueDescriptor elementwiseUnaryDescriptor;
129  elementwiseUnaryDescriptor.m_Parameters.m_Operation = UnaryOperation::Abs;
130 
131  return CreateElementwiseUnary(elementwiseUnaryDescriptor, info);
132 }
+
armnn::RefWorkloadFactory::CreateElementwiseUnary
std::unique_ptr< IWorkload > CreateElementwiseUnary(const ElementwiseUnaryQueueDescriptor &descriptor, const WorkloadInfo &info) const override
Definition: RefWorkloadFactory.cpp:270
+
armnn::IgnoreUnused
void IgnoreUnused(Ts &&...)
Definition: IgnoreUnused.hpp:14
+ +
+
+
+ +

◆ CreateActivation()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateActivation (const ActivationQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 134 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
136 {
137  return std::make_unique<RefActivationWorkload>(descriptor, info);
138 }
+
+
+
+ +

◆ CreateAddition()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateAddition (const AdditionQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 140 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
142 {
143  return std::make_unique<RefAdditionWorkload>(descriptor, info);
144 }
+
+
+
+ +

◆ CreateArgMinMax()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateArgMinMax (const ArgMinMaxQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 146 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
148 {
149  return std::make_unique<RefArgMinMaxWorkload>(descriptor, info);
150 }
+
+
+
+ +

◆ CreateBatchNormalization()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateBatchNormalization (const BatchNormalizationQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 152 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
155 {
156  return std::make_unique<RefBatchNormalizationWorkload>(descriptor, info);
157 }
+
+
+
+ +

◆ CreateBatchToSpaceNd()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateBatchToSpaceNd (const BatchToSpaceNdQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 159 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
161 {
162  return std::make_unique<RefBatchToSpaceNdWorkload>(descriptor, info);
163 }
+
+
+
+ +

◆ CreateComparison()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateComparison (const ComparisonQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 165 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateEqual(), RefWorkloadFactory::CreateGreater(), and RefWorkloadFactory::CreateSubTensorHandle().

+
167 {
168  return std::make_unique<RefComparisonWorkload>(descriptor, info);
169 }
+
+
+
+ +

◆ CreateConcat()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateConcat (const ConcatQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 171 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateMerger(), and RefWorkloadFactory::CreateSubTensorHandle().

+
173 {
174  return std::make_unique<RefConcatWorkload>(descriptor, info);
175 }
+
+
+
+ +

◆ CreateConstant()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateConstant (const ConstantQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 177 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
179 {
180  return std::make_unique<RefConstantWorkload>(descriptor, info);
181 }
+
+
+
+ +

◆ CreateConvertFp16ToFp32()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateConvertFp16ToFp32 (const ConvertFp16ToFp32QueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 183 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
186 {
187  return std::make_unique<RefConvertFp16ToFp32Workload>(descriptor, info);
188 }
+
+
+
+ +

◆ CreateConvertFp32ToFp16()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateConvertFp32ToFp16 (const ConvertFp32ToFp16QueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 190 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
193 {
194  return std::make_unique<RefConvertFp32ToFp16Workload>(descriptor, info);
195 }
+
+
+
+ +

◆ CreateConvolution2d()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateConvolution2d (const Convolution2dQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 197 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
199 {
200  return std::make_unique<RefConvolution2dWorkload>(descriptor, info);
201 }
+
+
+
+ +

◆ CreateDebug()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateDebug (const DebugQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 203 of file RefWorkloadFactory.cpp.

+ +

References armnn::info, armnn::IsBFloat16(), armnn::IsFloat16(), armnn::IsQAsymmS8(), armnn::IsQAsymmU8(), armnn::IsQSymmS16(), armnn::IsQSymmS8(), and armnn::IsSigned32().

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
205 {
206  if (IsBFloat16(info))
207  {
208  return std::make_unique<RefDebugBFloat16Workload>(descriptor, info);
209  }
210  if (IsFloat16(info))
211  {
212  return std::make_unique<RefDebugFloat16Workload>(descriptor, info);
213  }
214  if (IsQSymmS16(info))
215  {
216  return std::make_unique<RefDebugQSymmS16Workload>(descriptor, info);
217  }
218  if (IsQSymmS8(info))
219  {
220  return std::make_unique<RefDebugQSymmS8Workload>(descriptor, info);
221  }
222  if (IsQAsymmU8(info))
223  {
224  return std::make_unique<RefDebugQAsymmU8Workload>(descriptor, info);
225  }
226  if (IsQAsymmS8(info))
227  {
228  return std::make_unique<RefDebugQAsymmS8Workload>(descriptor, info);
229  }
230  if (IsSigned32(info))
231  {
232  return std::make_unique<RefDebugSigned32Workload>(descriptor, info);
233  }
234 
235  return MakeWorkload<RefDebugFloat32Workload, RefDebugQAsymmU8Workload>(descriptor, info);
236 }
armnn::IsQAsymmS8
bool IsQAsymmS8(const WorkloadInfo &info)
Definition: RefWorkloadFactory.cpp:73
+
armnn::IsQAsymmU8
bool IsQAsymmU8(const WorkloadInfo &info)
Definition: RefWorkloadFactory.cpp:78
+
armnn::IsQSymmS8
bool IsQSymmS8(const WorkloadInfo &info)
Definition: RefWorkloadFactory.cpp:68
+
armnn::IsBFloat16
bool IsBFloat16(const WorkloadInfo &info)
Definition: RefWorkloadFactory.cpp:53
+
armnn::IsFloat16
bool IsFloat16(const WorkloadInfo &info)
Definition: RefWorkloadFactory.cpp:58
+
armnn::IsSigned32
bool IsSigned32(const WorkloadInfo &info)
Definition: RefWorkloadFactory.cpp:48
+ +
armnn::IsQSymmS16
bool IsQSymmS16(const WorkloadInfo &info)
Definition: RefWorkloadFactory.cpp:63
+
+
+
+ +

◆ CreateDepthToSpace()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateDepthToSpace (const DepthToSpaceQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 238 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
240 {
241  return std::make_unique<RefDepthToSpaceWorkload>(descriptor, info);
242 }
+
+
+
+ +

◆ CreateDepthwiseConvolution2d()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateDepthwiseConvolution2d (const DepthwiseConvolution2dQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 244 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
247 {
248  return std::make_unique<RefDepthwiseConvolution2dWorkload>(descriptor, info);
249 }
+
+
+
+ +

◆ CreateDequantize()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateDequantize (const DequantizeQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 251 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
253 {
254  return std::make_unique<RefDequantizeWorkload>(descriptor, info);
255 }
+
+
+
+ +

◆ CreateDetectionPostProcess()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateDetectionPostProcess (const DetectionPostProcessQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 257 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
260 {
261  return std::make_unique<RefDetectionPostProcessWorkload>(descriptor, info);
262 }
+
+
+
+ +

◆ CreateDivision()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateDivision (const DivisionQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 264 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
266 {
267  return std::make_unique<RefDivisionWorkload>(descriptor, info);
268 }
+
+
+
+ +

◆ CreateElementwiseUnary()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateElementwiseUnary (const ElementwiseUnaryQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 270 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateAbs(), RefWorkloadFactory::CreateRsqrt(), and RefWorkloadFactory::CreateSubTensorHandle().

+
272 {
273  return std::make_unique<RefElementwiseUnaryWorkload>(descriptor, info);
274 }
+
+
+
+ +

◆ CreateEqual()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateEqual (const EqualQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 276 of file RefWorkloadFactory.cpp.

+ +

References RefWorkloadFactory::CreateComparison(), armnn::Equal, armnn::IgnoreUnused(), ComparisonDescriptor::m_Operation, and QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
278 {
279  IgnoreUnused(descriptor);
280  ComparisonQueueDescriptor comparisonDescriptor;
281  comparisonDescriptor.m_Parameters.m_Operation = ComparisonOperation::Equal;
282 
283  return CreateComparison(comparisonDescriptor, info);
284 }
+
armnn::RefWorkloadFactory::CreateComparison
std::unique_ptr< IWorkload > CreateComparison(const ComparisonQueueDescriptor &descriptor, const WorkloadInfo &info) const override
Definition: RefWorkloadFactory.cpp:165
+
armnn::IgnoreUnused
void IgnoreUnused(Ts &&...)
Definition: IgnoreUnused.hpp:14
+ +
+
+
+ +

◆ CreateFakeQuantization()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateFakeQuantization (const FakeQuantizationQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 286 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
289 {
290  return MakeWorkload<RefFakeQuantizationFloat32Workload, NullWorkload>(descriptor, info);
291 }
+
+
+
+ +

◆ CreateFloor()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateFloor (const FloorQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 293 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
295 {
296  return std::make_unique<RefFloorWorkload>(descriptor, info);
297 }
+
+
+
+ +

◆ CreateFullyConnected()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateFullyConnected (const FullyConnectedQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 299 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
302 {
303  return std::make_unique<RefFullyConnectedWorkload>(descriptor, info);
304 }
+
+
+
+ +

◆ CreateGather()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateGather (const GatherQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 306 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
308 {
309  return std::make_unique<RefGatherWorkload>(descriptor, info);
310 }
+
+
+
+ +

◆ CreateGreater()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateGreater (const GreaterQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 312 of file RefWorkloadFactory.cpp.

+ +

References RefWorkloadFactory::CreateComparison(), armnn::Greater, armnn::IgnoreUnused(), ComparisonDescriptor::m_Operation, and QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
314 {
315  IgnoreUnused(descriptor);
316  ComparisonQueueDescriptor comparisonDescriptor;
317  comparisonDescriptor.m_Parameters.m_Operation = ComparisonOperation::Greater;
318 
319  return CreateComparison(comparisonDescriptor, info);
320 }
+
armnn::RefWorkloadFactory::CreateComparison
std::unique_ptr< IWorkload > CreateComparison(const ComparisonQueueDescriptor &descriptor, const WorkloadInfo &info) const override
Definition: RefWorkloadFactory.cpp:165
+
armnn::IgnoreUnused
void IgnoreUnused(Ts &&...)
Definition: IgnoreUnused.hpp:14
+ +
+
+
+ +

◆ CreateInput()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateInput (const InputQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Implements IWorkloadFactory.

+ +

Definition at line 322 of file RefWorkloadFactory.cpp.

+ +

References armnn::info, WorkloadInfo::m_InputTensorInfos, and WorkloadInfo::m_OutputTensorInfos.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
324 {
325  if (info.m_InputTensorInfos.empty() )
326  {
327  throw InvalidArgumentException("RefWorkloadFactory::CreateInput: Input cannot be zero length");
328  }
329  if (info.m_OutputTensorInfos.empty())
330  {
331  throw InvalidArgumentException("RefWorkloadFactory::CreateInput: Output cannot be zero length");
332  }
333 
334  if (info.m_InputTensorInfos[0].GetNumBytes() != info.m_OutputTensorInfos[0].GetNumBytes())
335  {
336  throw InvalidArgumentException("RefWorkloadFactory::CreateInput: data input and output differ in byte count.");
337  }
338 
339  return std::make_unique<CopyMemGenericWorkload>(descriptor, info);
340 }
+
+
+
+ +

◆ CreateInstanceNormalization()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateInstanceNormalization (const InstanceNormalizationQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 342 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
345 {
346  return std::make_unique<RefInstanceNormalizationWorkload>(descriptor, info);
347 }
+
+
+
+ +

◆ CreateL2Normalization()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateL2Normalization (const L2NormalizationQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 349 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
351 {
352  return std::make_unique<RefL2NormalizationWorkload>(descriptor, info);
353 }
+
+
+
+ +

◆ CreateLogSoftmax()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateLogSoftmax (const LogSoftmaxQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 355 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
357 {
358  return std::make_unique<RefLogSoftmaxWorkload>(descriptor, info);
359 }
+
+
+
+ +

◆ CreateLstm()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateLstm (const LstmQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 361 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
363 {
364  return std::make_unique<RefLstmWorkload>(descriptor, info);
365 }
+
+
+
+ +

◆ CreateMaximum()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateMaximum (const MaximumQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 367 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
369 {
370  return std::make_unique<RefMaximumWorkload>(descriptor, info);
371 }
+
+
+
+ +

◆ CreateMean()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateMean (const MeanQueueDescriptor & descriptor,
const WorkloadInfo & Info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 373 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
375 {
376  return std::make_unique<RefMeanWorkload>(descriptor, info);
377 }
+
+
+
+ +

◆ CreateMemCopy()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateMemCopy (const MemCopyQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 379 of file RefWorkloadFactory.cpp.

+ +

References armnn::info, and QueueDescriptor::m_Inputs.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
381 {
382  if (descriptor.m_Inputs.empty())
383  {
384  throw InvalidArgumentException("RefWorkloadFactory: CreateMemCopy() expected an input tensor.");
385  }
386  return std::make_unique<CopyMemGenericWorkload>(descriptor, info);
387 }
+
+
+
+ +

◆ CreateMemImport()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateMemImport (const MemImportQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 389 of file RefWorkloadFactory.cpp.

+ +

References armnn::info, and QueueDescriptor::m_Inputs.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
391 {
392  if (descriptor.m_Inputs.empty())
393  {
394  throw InvalidArgumentException("RefWorkloadFactory: CreateMemImport() expected an input tensor.");
395  }
396  return std::make_unique<ImportMemGenericWorkload>(descriptor, info);
397 }
+
+
+
+ +

◆ CreateMerger()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateMerger (const MergerQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 399 of file RefWorkloadFactory.cpp.

+ +

References RefWorkloadFactory::CreateConcat().

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
401 {
402  return CreateConcat(descriptor, info);
403 }
armnn::RefWorkloadFactory::CreateConcat
std::unique_ptr< IWorkload > CreateConcat(const ConcatQueueDescriptor &descriptor, const WorkloadInfo &info) const override
Definition: RefWorkloadFactory.cpp:171
+ +
+
+
+ +

◆ CreateMinimum()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateMinimum (const MinimumQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 405 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
407 {
408  return std::make_unique<RefMinimumWorkload>(descriptor, info);
409 }
+
+
+
+ +

◆ CreateMultiplication()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateMultiplication (const MultiplicationQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 411 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
413 {
414  return std::make_unique<RefMultiplicationWorkload>(descriptor, info);
415 }
+
+
+
+ +

◆ CreateNormalization()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateNormalization (const NormalizationQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 417 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
419 {
420  return std::make_unique<RefNormalizationWorkload>(descriptor, info);
421 }
+
+
+
+ +

◆ CreateOutput()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateOutput (const OutputQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 423 of file RefWorkloadFactory.cpp.

+ +

References armnn::info, WorkloadInfo::m_InputTensorInfos, and WorkloadInfo::m_OutputTensorInfos.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
425 {
426  if (info.m_InputTensorInfos.empty() )
427  {
428  throw InvalidArgumentException("RefWorkloadFactory::CreateOutput: Input cannot be zero length");
429  }
430  if (info.m_OutputTensorInfos.empty())
431  {
432  throw InvalidArgumentException("RefWorkloadFactory::CreateOutput: Output cannot be zero length");
433  }
434  if (info.m_InputTensorInfos[0].GetNumBytes() != info.m_OutputTensorInfos[0].GetNumBytes())
435  {
436  throw InvalidArgumentException("RefWorkloadFactory::CreateOutput: data input and output differ in byte count.");
437  }
438 
439  return std::make_unique<CopyMemGenericWorkload>(descriptor, info);
440 }
+
+
+
+ +

◆ CreatePad()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreatePad (const PadQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 442 of file RefWorkloadFactory.cpp.

+ +

References armnn::info, armnn::IsBFloat16(), armnn::IsFloat16(), and armnn::IsQSymmS16().

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
444 {
445  if (IsQSymmS16(info))
446  {
447  return std::make_unique<RefPadQSymm16Workload>(descriptor, info);
448  }
449  else if (IsFloat16(info))
450  {
451  return std::make_unique<RefPadFloat16Workload>(descriptor, info);
452  }
453  else if (IsBFloat16(info))
454  {
455  return std::make_unique<RefPadBFloat16Workload>(descriptor, info);
456  }
457  return MakeWorkload<RefPadFloat32Workload, RefPadQAsymm8Workload>(descriptor, info);
458 }
armnn::IsBFloat16
bool IsBFloat16(const WorkloadInfo &info)
Definition: RefWorkloadFactory.cpp:53
+
armnn::IsFloat16
bool IsFloat16(const WorkloadInfo &info)
Definition: RefWorkloadFactory.cpp:58
+ +
armnn::IsQSymmS16
bool IsQSymmS16(const WorkloadInfo &info)
Definition: RefWorkloadFactory.cpp:63
+
+
+
+ +

◆ CreatePermute()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreatePermute (const PermuteQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 460 of file RefWorkloadFactory.cpp.

+ +

References armnn::info, armnn::IsBFloat16(), and armnn::IsQSymmS16().

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
462 {
463  if (IsQSymmS16(info))
464  {
465  return std::make_unique<RefPermuteQSymm16Workload>(descriptor, info);
466  }
467  else if (IsBFloat16(info))
468  {
469  return std::make_unique<RefPermuteBFloat16Workload>(descriptor, info);
470  }
471  return MakeWorkloadHelper<RefPermuteFloat16Workload, RefPermuteFloat32Workload, RefPermuteQAsymm8Workload,
472  NullWorkload, NullWorkload, NullWorkload>(descriptor, info);
473 }
armnn::RefPermuteFloat16Workload
RefPermuteWorkload< DataType::Float16 > RefPermuteFloat16Workload
Definition: RefPermuteWorkload.hpp:31
+
armnn::RefPermuteFloat32Workload
RefPermuteWorkload< DataType::Float32 > RefPermuteFloat32Workload
Definition: RefPermuteWorkload.hpp:32
+
armnn::IsBFloat16
bool IsBFloat16(const WorkloadInfo &info)
Definition: RefWorkloadFactory.cpp:53
+ +
armnn::RefPermuteQAsymm8Workload
RefPermuteWorkload< DataType::QAsymmU8 > RefPermuteQAsymm8Workload
Definition: RefPermuteWorkload.hpp:33
+
armnn::IsQSymmS16
bool IsQSymmS16(const WorkloadInfo &info)
Definition: RefWorkloadFactory.cpp:63
+
+
+
+ +

◆ CreatePooling2d()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreatePooling2d (const Pooling2dQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 475 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
477 {
478  return std::make_unique<RefPooling2dWorkload>(descriptor, info);
479 }
+
+
+
+ +

◆ CreatePreCompiled()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreatePreCompiled (const PreCompiledQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 481 of file RefWorkloadFactory.cpp.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
483 {
484  return nullptr;
485 }
+
+
+ +

◆ CreatePrelu()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreatePrelu (const PreluQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 487 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
489 {
490  return std::make_unique<RefPreluWorkload>(descriptor, info);
491 }
+
+
+
+ +

◆ CreateQuantize()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateQuantize (const QuantizeQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 493 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
495 {
496  return std::make_unique<RefQuantizeWorkload>(descriptor, info);
497 }
+
+
+
+ +

◆ CreateReshape()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateReshape (const ReshapeQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 499 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
501 {
502  return std::make_unique<RefReshapeWorkload>(descriptor, info);
503 }
+
+
+
+ +

◆ CreateResize()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateResize (const ResizeQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 505 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateResizeBilinear(), and RefWorkloadFactory::CreateSubTensorHandle().

+
507 {
508  return std::make_unique<RefResizeWorkload>(descriptor, info);
509 }
+
+
+
+ +

◆ CreateResizeBilinear()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateResizeBilinear (const ResizeBilinearQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 511 of file RefWorkloadFactory.cpp.

+ +

References armnn::Bilinear, RefWorkloadFactory::CreateResize(), ResizeBilinearDescriptor::m_DataLayout, ResizeDescriptor::m_DataLayout, ResizeDescriptor::m_Method, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, ResizeBilinearDescriptor::m_TargetHeight, ResizeDescriptor::m_TargetHeight, ResizeBilinearDescriptor::m_TargetWidth, and ResizeDescriptor::m_TargetWidth.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
513 {
514  ResizeQueueDescriptor resizeDescriptor;
515  resizeDescriptor.m_Parameters.m_Method = ResizeMethod::Bilinear;
516  resizeDescriptor.m_Parameters.m_DataLayout = descriptor.m_Parameters.m_DataLayout;
517  resizeDescriptor.m_Parameters.m_TargetWidth = descriptor.m_Parameters.m_TargetWidth;
518  resizeDescriptor.m_Parameters.m_TargetHeight = descriptor.m_Parameters.m_TargetHeight;
519 
520  return CreateResize(resizeDescriptor, info);
521 }
armnn::RefWorkloadFactory::CreateResize
std::unique_ptr< IWorkload > CreateResize(const ResizeQueueDescriptor &descriptor, const WorkloadInfo &info) const override
Definition: RefWorkloadFactory.cpp:505
+ + +
+
+
+ +

◆ CreateRsqrt()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateRsqrt (const RsqrtQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 523 of file RefWorkloadFactory.cpp.

+ +

References RefWorkloadFactory::CreateElementwiseUnary(), armnn::IgnoreUnused(), ElementwiseUnaryDescriptor::m_Operation, QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters, and armnn::Rsqrt.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
525 {
526  IgnoreUnused(descriptor);
527  ElementwiseUnaryQueueDescriptor elementwiseUnaryDescriptor;
528  elementwiseUnaryDescriptor.m_Parameters.m_Operation = UnaryOperation::Rsqrt;
529 
530  return CreateElementwiseUnary(elementwiseUnaryDescriptor, info);
531 }
armnn::RefWorkloadFactory::CreateElementwiseUnary
std::unique_ptr< IWorkload > CreateElementwiseUnary(const ElementwiseUnaryQueueDescriptor &descriptor, const WorkloadInfo &info) const override
Definition: RefWorkloadFactory.cpp:270
+
armnn::IgnoreUnused
void IgnoreUnused(Ts &&...)
Definition: IgnoreUnused.hpp:14
+ + +
+
+
+ +

◆ CreateSlice()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateSlice (const SliceQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 533 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
535 {
536  return std::make_unique<RefSliceWorkload>(descriptor, info);
537 }
+
+
+
+ +

◆ CreateSoftmax()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateSoftmax (const SoftmaxQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 539 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
541 {
542  return std::make_unique<RefSoftmaxWorkload>(descriptor, info);
543 }
+
+
+
+ +

◆ CreateSpaceToBatchNd()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateSpaceToBatchNd (const SpaceToBatchNdQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 545 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
547 {
548  return std::make_unique<RefSpaceToBatchNdWorkload>(descriptor, info);
549 }
+
+
+
+ +

◆ CreateSpaceToDepth()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateSpaceToDepth (const SpaceToDepthQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 551 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
553 {
554  return std::make_unique<RefSpaceToDepthWorkload>(descriptor, info);
555 }
+
+
+
+ +

◆ CreateSplitter()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateSplitter (const SplitterQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 557 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
559 {
560  return std::make_unique<RefSplitterWorkload>(descriptor, info);
561 }
+
+
+
+ +

◆ CreateStack()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateStack (const StackQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 563 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
565 {
566  return std::make_unique<RefStackWorkload>(descriptor, info);
567 }
+
+
+
+ +

◆ CreateStridedSlice()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateStridedSlice (const StridedSliceQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 569 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
571 {
572  return std::make_unique<RefStridedSliceWorkload>(descriptor, info);
573 }
+
+
+
+ +

◆ CreateSubTensorHandle()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + + + + + + + +
std::unique_ptr<ITensorHandle> CreateSubTensorHandle (ITensorHandle & parent,
TensorShape const & subTensorShape,
unsigned int const * subTensorOrigin 
) const
+ 		+inlineoverridevirtual
+
+ +

Implements IWorkloadFactory.

+ +

Definition at line 46 of file RefWorkloadFactory.hpp.

+ +

References ARMNN_DEPRECATED_MSG, RefWorkloadFactory::CreateAbs(), RefWorkloadFactory::CreateActivation(), RefWorkloadFactory::CreateAddition(), RefWorkloadFactory::CreateArgMinMax(), RefWorkloadFactory::CreateBatchNormalization(), RefWorkloadFactory::CreateBatchToSpaceNd(), RefWorkloadFactory::CreateComparison(), RefWorkloadFactory::CreateConcat(), RefWorkloadFactory::CreateConstant(), RefWorkloadFactory::CreateConvertFp16ToFp32(), RefWorkloadFactory::CreateConvertFp32ToFp16(), RefWorkloadFactory::CreateConvolution2d(), RefWorkloadFactory::CreateDebug(), RefWorkloadFactory::CreateDepthToSpace(), RefWorkloadFactory::CreateDepthwiseConvolution2d(), RefWorkloadFactory::CreateDequantize(), RefWorkloadFactory::CreateDetectionPostProcess(), RefWorkloadFactory::CreateDivision(), RefWorkloadFactory::CreateElementwiseUnary(), RefWorkloadFactory::CreateEqual(), RefWorkloadFactory::CreateFakeQuantization(), RefWorkloadFactory::CreateFloor(), RefWorkloadFactory::CreateFullyConnected(), RefWorkloadFactory::CreateGather(), RefWorkloadFactory::CreateGreater(), RefWorkloadFactory::CreateInput(), RefWorkloadFactory::CreateInstanceNormalization(), RefWorkloadFactory::CreateL2Normalization(), RefWorkloadFactory::CreateLogSoftmax(), RefWorkloadFactory::CreateLstm(), RefWorkloadFactory::CreateMaximum(), RefWorkloadFactory::CreateMean(), RefWorkloadFactory::CreateMemCopy(), RefWorkloadFactory::CreateMemImport(), RefWorkloadFactory::CreateMerger(), RefWorkloadFactory::CreateMinimum(), RefWorkloadFactory::CreateMultiplication(), RefWorkloadFactory::CreateNormalization(), RefWorkloadFactory::CreateOutput(), RefWorkloadFactory::CreatePad(), RefWorkloadFactory::CreatePermute(), RefWorkloadFactory::CreatePooling2d(), RefWorkloadFactory::CreatePreCompiled(), RefWorkloadFactory::CreatePrelu(), RefWorkloadFactory::CreateQuantize(), RefWorkloadFactory::CreateReshape(), RefWorkloadFactory::CreateResize(), RefWorkloadFactory::CreateResizeBilinear(), RefWorkloadFactory::CreateRsqrt(), RefWorkloadFactory::CreateSlice(), RefWorkloadFactory::CreateSoftmax(), RefWorkloadFactory::CreateSpaceToBatchNd(), RefWorkloadFactory::CreateSpaceToDepth(), RefWorkloadFactory::CreateSplitter(), RefWorkloadFactory::CreateStack(), RefWorkloadFactory::CreateStridedSlice(), RefWorkloadFactory::CreateSubtraction(), RefWorkloadFactory::CreateTensorHandle(), RefWorkloadFactory::CreateTranspose(), RefWorkloadFactory::CreateTransposeConvolution2d(), armnn::IgnoreUnused(), armnn::Info, and armnn::info.

+
49  {
50  IgnoreUnused(parent, subTensorShape, subTensorOrigin);
51  return nullptr;
52  }
armnn::IgnoreUnused
void IgnoreUnused(Ts &&...)
Definition: IgnoreUnused.hpp:14
+
+
+
+ +

◆ CreateSubtraction()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateSubtraction (const SubtractionQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 575 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
577 {
578  return std::make_unique<RefSubtractionWorkload>(descriptor, info);
579 }
+
+
+
+ +

◆ CreateTensorHandle() [1/2]

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< ITensorHandle > CreateTensorHandle (const TensorInfo & tensorInfo,
const bool IsMemoryManaged = true 
) const
+ 		+overridevirtual
+
+ +

Implements IWorkloadFactory.

+ +

Definition at line 105 of file RefWorkloadFactory.cpp.

+ +

References armnn::IgnoreUnused().

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
107 {
108  // For Ref it is okay to make the TensorHandle memory managed as it can also store a pointer
109  // to unmanaged memory. This also ensures memory alignment.
110  IgnoreUnused(isMemoryManaged);
111  return std::make_unique<RefTensorHandle>(tensorInfo, m_MemoryManager);
112 }
armnn::IgnoreUnused
void IgnoreUnused(Ts &&...)
Definition: IgnoreUnused.hpp:14
+
+
+
+ +

◆ CreateTensorHandle() [2/2]

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + + + + + + + +
std::unique_ptr< ITensorHandle > CreateTensorHandle (const TensorInfo & tensorInfo,
DataLayout dataLayout,
const bool IsMemoryManaged = true 
) const
+ 		+overridevirtual
+
+ +

Implements IWorkloadFactory.

+ +

Definition at line 114 of file RefWorkloadFactory.cpp.

+ +

References armnn::IgnoreUnused().

+
117 {
118  // For Ref it is okay to make the TensorHandle memory managed as it can also store a pointer
119  // to unmanaged memory. This also ensures memory alignment.
120  IgnoreUnused(isMemoryManaged, dataLayout);
121  return std::make_unique<RefTensorHandle>(tensorInfo, m_MemoryManager);
122 }
armnn::IgnoreUnused
void IgnoreUnused(Ts &&...)
Definition: IgnoreUnused.hpp:14
+
+
+
+ +

◆ CreateTranspose()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateTranspose (const TransposeQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 581 of file RefWorkloadFactory.cpp.

+ +

References armnn::info, armnn::IsBFloat16(), and armnn::IsQSymmS16().

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
583 {
584  if (IsQSymmS16(info))
585  {
586  return std::make_unique<RefTransposeQSymm16Workload>(descriptor, info);
587  }
588  else if (IsBFloat16(info))
589  {
590  return std::make_unique<RefTransposeBFloat16Workload>(descriptor, info);
591  }
592  return MakeWorkloadHelper<RefTransposeFloat16Workload, RefTransposeFloat32Workload, RefTransposeQAsymm8Workload,
593  NullWorkload, NullWorkload, NullWorkload>(descriptor, info);
594 }
armnn::RefTransposeFloat16Workload
RefTransposeWorkload< DataType::Float16 > RefTransposeFloat16Workload
Definition: RefTransposeWorkload.hpp:31
+
armnn::IsBFloat16
bool IsBFloat16(const WorkloadInfo &info)
Definition: RefWorkloadFactory.cpp:53
+
armnn::RefTransposeFloat32Workload
RefTransposeWorkload< DataType::Float32 > RefTransposeFloat32Workload
Definition: RefTransposeWorkload.hpp:32
+
armnn::RefTransposeQAsymm8Workload
RefTransposeWorkload< DataType::QAsymmU8 > RefTransposeQAsymm8Workload
Definition: RefTransposeWorkload.hpp:33
+ +
armnn::IsQSymmS16
bool IsQSymmS16(const WorkloadInfo &info)
Definition: RefWorkloadFactory.cpp:63
+
+
+
+ +

◆ CreateTransposeConvolution2d()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + +
std::unique_ptr< IWorkload > CreateTransposeConvolution2d (const TransposeConvolution2dQueueDescriptor & descriptor,
const WorkloadInfo & info 
) const
+ 		+overridevirtual
+
+ +

Reimplemented from IWorkloadFactory.

+ +

Definition at line 596 of file RefWorkloadFactory.cpp.

+ +

References armnn::info.

+ +

Referenced by RefWorkloadFactory::CreateSubTensorHandle().

+
599 {
600  return std::make_unique<RefTransposeConvolution2dWorkload>(descriptor, info);
601 }
+
+
+
+ +

◆ GetBackendId()

+ +
+
+ + + + + +
+ + + + + + + +
const BackendId & GetBackendId () const
+ 		+overridevirtual
+
+ +

Implements IWorkloadFactory.

+ +

Definition at line 93 of file RefWorkloadFactory.cpp.

+ +

Referenced by RefWorkloadFactory::~RefWorkloadFactory().

+
94 {
95  return s_Id;
96 }
+
+
+ +

◆ IsLayerSupported()

+ +
+
+ + + + + +
+ + + + + + + + + + + + + + + + + + + + + + + + +
bool IsLayerSupported (const Layer & layer,
Optional< DataType > dataType,
std::string & outReasonIfUnsupported 
)
+ 		+static
+
+ +

Definition at line 98 of file RefWorkloadFactory.cpp.

+ +

References IWorkloadFactory::IsLayerSupported().

+ +

Referenced by RefWorkloadFactory::~RefWorkloadFactory().

+
101 {
102  return IWorkloadFactory::IsLayerSupported(s_Id, layer, dataType, outReasonIfUnsupported);
103 }
armnn::IWorkloadFactory::IsLayerSupported
static bool IsLayerSupported(const BackendId &backendId, const IConnectableLayer &layer, Optional< DataType > dataType, std::string &outReasonIfUnsupported)
Definition: WorkloadFactory.cpp:45
+
+
+
+ +

◆ SupportsSubTensors()

+ +
+
+ + + + + +
+ + + + + + + +
bool SupportsSubTensors () const
+ 		+inlineoverridevirtual
+
+ +

Implements IWorkloadFactory.

+ +

Definition at line 44 of file RefWorkloadFactory.hpp.

+
44 { return false; }
+
+
+
The documentation for this class was generated from the following files: +
+
+ + + + -- cgit v1.2.1