Lines Matching refs:descriptor
113 std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateAbs(const AbsQueueDescriptor& descriptor, in CreateAbs() argument
116 IgnoreUnused(descriptor); in CreateAbs()
124 …e_ptr<IWorkload> NeonWorkloadFactory::CreateActivation(const ActivationQueueDescriptor& descriptor, in CreateActivation() argument
127 return std::make_unique<NeonActivationWorkload>(descriptor, info); in CreateActivation()
130 …tr<armnn::IWorkload> NeonWorkloadFactory::CreateAddition(const AdditionQueueDescriptor& descriptor, in CreateAddition() argument
133 return std::make_unique<NeonAdditionWorkload>(descriptor, info); in CreateAddition()
136 …que_ptr<IWorkload> NeonWorkloadFactory::CreateArgMinMax(const ArgMinMaxQueueDescriptor& descriptor, in CreateArgMinMax() argument
139 return std::make_unique<NeonArgMinMaxWorkload>(descriptor, info); in CreateArgMinMax()
143 const BatchNormalizationQueueDescriptor& descriptor, const WorkloadInfo& info) const in CreateBatchNormalization() argument
145 return std::make_unique<NeonBatchNormalizationWorkload>(descriptor, info); in CreateBatchNormalization()
148 …orkload> NeonWorkloadFactory::CreateBatchToSpaceNd(const BatchToSpaceNdQueueDescriptor& descriptor, in CreateBatchToSpaceNd() argument
151 return std::make_unique<NeonBatchToSpaceNdWorkload>(descriptor, info); in CreateBatchToSpaceNd()
154 …e_ptr<IWorkload> NeonWorkloadFactory::CreateComparison(const ComparisonQueueDescriptor& descriptor, in CreateComparison() argument
157 return std::make_unique<NeonComparisonWorkload>(descriptor, info); in CreateComparison()
160 …ue_ptr<armnn::IWorkload> NeonWorkloadFactory::CreateConcat(const ConcatQueueDescriptor& descriptor, in CreateConcat() argument
163 return std::make_unique<NeonConcatWorkload>(descriptor, info); in CreateConcat()
166 …nique_ptr<IWorkload> NeonWorkloadFactory::CreateConstant(const ConstantQueueDescriptor& descriptor, in CreateConstant() argument
169 return std::make_unique<NeonConstantWorkload>(descriptor, info); in CreateConstant()
173 const ConvertBf16ToFp32QueueDescriptor& descriptor, in CreateConvertBf16ToFp32() argument
176 return std::make_unique<NeonConvertBf16ToFp32Workload>(descriptor, info); in CreateConvertBf16ToFp32()
180 const ConvertFp16ToFp32QueueDescriptor& descriptor, in CreateConvertFp16ToFp32() argument
183 return std::make_unique<NeonConvertFp16ToFp32Workload>(descriptor, info); in CreateConvertFp16ToFp32()
187 const ConvertFp32ToBf16QueueDescriptor& descriptor, in CreateConvertFp32ToBf16() argument
190 return std::make_unique<NeonConvertFp32ToBf16Workload>(descriptor, info); in CreateConvertFp32ToBf16()
194 const ConvertFp32ToFp16QueueDescriptor& descriptor, in CreateConvertFp32ToFp16() argument
197 return std::make_unique<NeonConvertFp32ToFp16Workload>(descriptor, info); in CreateConvertFp32ToFp16()
201 const Convolution2dQueueDescriptor& descriptor, const WorkloadInfo& info) const in CreateConvolution2d() argument
215 return std::make_unique<NeonConvolution2dWorkload>(descriptor, in CreateConvolution2d()
221 std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateDebug(const DebugQueueDescriptor& descriptor, in CreateDebug() argument
224 return MakeWorkloadHelper<NullWorkload, NullWorkload>(descriptor, info); in CreateDebug()
227 …r<IWorkload> NeonWorkloadFactory::CreateDepthToSpace(const DepthToSpaceQueueDescriptor& descriptor, in CreateDepthToSpace() argument
230 return std::make_unique<NeonDepthToSpaceWorkload>(descriptor, info); in CreateDepthToSpace()
234 const DepthwiseConvolution2dQueueDescriptor& descriptor, const WorkloadInfo& info) const in CreateDepthwiseConvolution2d() argument
236 return std::make_unique<NeonDepthwiseConvolutionWorkload>(descriptor, info); in CreateDepthwiseConvolution2d()
239 …e_ptr<IWorkload> NeonWorkloadFactory::CreateDequantize(const DequantizeQueueDescriptor& descriptor, in CreateDequantize() argument
242 return std::make_unique<NeonDequantizeWorkload>(descriptor, info); in CreateDequantize()
246 …const armnn::DetectionPostProcessQueueDescriptor& descriptor, const armnn::WorkloadInfo& info) con… in CreateDetectionPostProcess() argument
248 return MakeWorkloadHelper<NullWorkload, NullWorkload>(descriptor, info); in CreateDetectionPostProcess()
252 const DivisionQueueDescriptor& descriptor, const WorkloadInfo& info) const in CreateDivision() argument
254 return std::make_unique<NeonDivisionWorkload>(descriptor, info); in CreateDivision()
258 const ElementwiseUnaryQueueDescriptor& descriptor, const WorkloadInfo& info) const in CreateElementwiseUnary() argument
260 switch(descriptor.m_Parameters.m_Operation) in CreateElementwiseUnary()
265 absQueueDescriptor.m_Inputs = descriptor.m_Inputs; in CreateElementwiseUnary()
266 absQueueDescriptor.m_Outputs = descriptor.m_Outputs; in CreateElementwiseUnary()
273 rsqrtQueueDescriptor.m_Inputs = descriptor.m_Inputs; in CreateElementwiseUnary()
274 rsqrtQueueDescriptor.m_Outputs = descriptor.m_Outputs; in CreateElementwiseUnary()
279 return std::make_unique<NeonNegWorkload>(descriptor, info); in CreateElementwiseUnary()
281 return std::make_unique<NeonExpWorkload>(descriptor, info); in CreateElementwiseUnary()
283 return std::make_unique<NeonLogicalNotWorkload>(descriptor, info); in CreateElementwiseUnary()
289 std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateEqual(const EqualQueueDescriptor& descriptor, in CreateEqual() argument
292 IgnoreUnused(descriptor); in CreateEqual()
300 std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateFill(const FillQueueDescriptor& descriptor, in CreateFill() argument
303 return std::make_unique<NeonFillWorkload>(descriptor, info); in CreateFill()
306 std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateFloor(const FloorQueueDescriptor& descriptor, in CreateFloor() argument
309 return MakeWorkloadHelper<NeonFloorFloatWorkload, NullWorkload>(descriptor, info); in CreateFloor()
313 const FullyConnectedQueueDescriptor& descriptor, const WorkloadInfo& info) const in CreateFullyConnected() argument
315 …return std::make_unique<NeonFullyConnectedWorkload>(descriptor, info, m_MemoryManager->GetIntraLay… in CreateFullyConnected()
318 …ue_ptr<IWorkload> NeonWorkloadFactory::CreateGather(const armnn::GatherQueueDescriptor& descriptor, in CreateGather() argument
321 return std::make_unique<NeonGatherWorkload>(descriptor, info); in CreateGather()
324 …:unique_ptr<IWorkload> NeonWorkloadFactory::CreateGreater(const GreaterQueueDescriptor& descriptor, in CreateGreater() argument
327 IgnoreUnused(descriptor); in CreateGreater()
335 std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateInput(const InputQueueDescriptor& descriptor, in CreateInput() argument
338 return std::make_unique<CopyMemGenericWorkload>(descriptor, info); in CreateInput()
342 const InstanceNormalizationQueueDescriptor& descriptor, in CreateInstanceNormalization() argument
345 return std::make_unique<NeonInstanceNormalizationWorkload>(descriptor, info); in CreateInstanceNormalization()
348 …kload> NeonWorkloadFactory::CreateL2Normalization(const L2NormalizationQueueDescriptor& descriptor, in CreateL2Normalization() argument
351 return MakeWorkloadHelper<NeonL2NormalizationFloatWorkload, NullWorkload>(descriptor, info, in CreateL2Normalization()
355 …e_ptr<IWorkload> NeonWorkloadFactory::CreateLogSoftmax(const LogSoftmaxQueueDescriptor& descriptor, in CreateLogSoftmax() argument
358 …return std::make_unique<NeonLogSoftmaxWorkload>(descriptor, info, m_MemoryManager->GetIntraLayerMa… in CreateLogSoftmax()
361 …IWorkload> NeonWorkloadFactory::CreateLogicalBinary(const LogicalBinaryQueueDescriptor& descriptor, in CreateLogicalBinary() argument
364 switch(descriptor.m_Parameters.m_Operation) in CreateLogicalBinary()
367 return std::make_unique<NeonLogicalAndWorkload>(descriptor, info); in CreateLogicalBinary()
369 return std::make_unique<NeonLogicalOrWorkload>(descriptor, info); in CreateLogicalBinary()
375 std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateLstm(const LstmQueueDescriptor& descriptor, in CreateLstm() argument
378 return MakeWorkloadHelper<NeonLstmFloatWorkload, NullWorkload>(descriptor, info); in CreateLstm()
381 …:unique_ptr<IWorkload> NeonWorkloadFactory::CreateMaximum(const MaximumQueueDescriptor& descriptor, in CreateMaximum() argument
384 return std::make_unique<NeonMaximumWorkload>(descriptor, info); in CreateMaximum()
387 std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateMean(const MeanQueueDescriptor& descriptor, in CreateMean() argument
390 return std::make_unique<NeonMeanWorkload>(descriptor, info); in CreateMean()
393 …_ptr<armnn::IWorkload> NeonWorkloadFactory::CreateMemCopy(const MemCopyQueueDescriptor& descriptor, in CreateMemCopy() argument
396 if (descriptor.m_Inputs.empty() || !descriptor.m_Inputs[0]) in CreateMemCopy()
401 return MakeWorkloadHelper<CopyMemGenericWorkload, CopyMemGenericWorkload>(descriptor, info); in CreateMemCopy()
404 …<armnn::IWorkload> NeonWorkloadFactory::CreateMemImport(const MemImportQueueDescriptor& descriptor, in CreateMemImport() argument
407 if (descriptor.m_Inputs.empty() || !descriptor.m_Inputs[0]) in CreateMemImport()
412 return std::make_unique<ImportMemGenericWorkload>(descriptor, info); in CreateMemImport()
415 …ue_ptr<armnn::IWorkload> NeonWorkloadFactory::CreateMerger(const MergerQueueDescriptor& descriptor, in CreateMerger() argument
418 return CreateConcat(descriptor, info); in CreateMerger()
421 …:unique_ptr<IWorkload> NeonWorkloadFactory::CreateMinimum(const MinimumQueueDescriptor& descriptor, in CreateMinimum() argument
424 return std::make_unique<NeonMinimumWorkload>(descriptor, info); in CreateMinimum()
428 const MultiplicationQueueDescriptor& descriptor, const WorkloadInfo& info) const in CreateMultiplication() argument
430 return std::make_unique<NeonMultiplicationWorkload>(descriptor, info); in CreateMultiplication()
434 const NormalizationQueueDescriptor& descriptor, const WorkloadInfo& info) const in CreateNormalization() argument
436 return MakeWorkloadHelper<NeonNormalizationFloatWorkload, NullWorkload>(descriptor, info, in CreateNormalization()
440 …d::unique_ptr<IWorkload> NeonWorkloadFactory::CreateOutput(const OutputQueueDescriptor& descriptor, in CreateOutput() argument
443 return std::make_unique<CopyMemGenericWorkload>(descriptor, info); in CreateOutput()
446 std::unique_ptr<IWorkload> NeonWorkloadFactory::CreatePad(const PadQueueDescriptor& descriptor, in CreatePad() argument
449 return std::make_unique<NeonPadWorkload>(descriptor, info); in CreatePad()
452 …_ptr<armnn::IWorkload> NeonWorkloadFactory::CreatePermute(const PermuteQueueDescriptor& descriptor, in CreatePermute() argument
455 return std::make_unique<NeonPermuteWorkload>(descriptor, info); in CreatePermute()
458 …<armnn::IWorkload> NeonWorkloadFactory::CreatePooling2d(const Pooling2dQueueDescriptor& descriptor, in CreatePooling2d() argument
461 return std::make_unique<NeonPooling2dWorkload>(descriptor, info); in CreatePooling2d()
464 …ptr<IWorkload> NeonWorkloadFactory::CreatePreCompiled(const PreCompiledQueueDescriptor& descriptor, in CreatePreCompiled() argument
467 return MakeWorkloadHelper<NullWorkload, NullWorkload>(descriptor, info); in CreatePreCompiled()
470 …r<armnn::IWorkload> NeonWorkloadFactory::CreatePrelu(const armnn::PreluQueueDescriptor &descriptor, in CreatePrelu() argument
473 return std::make_unique<NeonPreluWorkload>(descriptor, info); in CreatePrelu()
476 std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateQLstm(const QLstmQueueDescriptor& descriptor, in CreateQLstm() argument
479 return std::make_unique<NeonQLstmWorkload>(descriptor, info); in CreateQLstm()
482 …tr<armnn::IWorkload> NeonWorkloadFactory::CreateQuantize(const QuantizeQueueDescriptor& descriptor, in CreateQuantize() argument
485 return std::make_unique<NeonQuantizeWorkload>(descriptor, info); in CreateQuantize()
488 …IWorkload> NeonWorkloadFactory::CreateQuantizedLstm(const QuantizedLstmQueueDescriptor& descriptor, in CreateQuantizedLstm() argument
491 return std::make_unique<NeonQuantizedLstmWorkload>(descriptor, info); in CreateQuantizedLstm()
494 …:unique_ptr<IWorkload> NeonWorkloadFactory::CreateReshape(const ReshapeQueueDescriptor& descriptor, in CreateReshape() argument
497 return std::make_unique<NeonReshapeWorkload>(descriptor, info); in CreateReshape()
500 …d::unique_ptr<IWorkload> NeonWorkloadFactory::CreateResize(const ResizeQueueDescriptor& descriptor, in CreateResize() argument
503 return std::make_unique<NeonResizeWorkload>(descriptor, info); in CreateResize()
507 const ResizeBilinearQueueDescriptor& descriptor, in CreateResizeBilinear() argument
511 resizeDescriptor.m_Inputs = descriptor.m_Inputs; in CreateResizeBilinear()
512 resizeDescriptor.m_Outputs = descriptor.m_Outputs; in CreateResizeBilinear()
514 resizeDescriptor.m_Parameters.m_DataLayout = descriptor.m_Parameters.m_DataLayout; in CreateResizeBilinear()
515 resizeDescriptor.m_Parameters.m_TargetWidth = descriptor.m_Parameters.m_TargetWidth; in CreateResizeBilinear()
516 resizeDescriptor.m_Parameters.m_TargetHeight = descriptor.m_Parameters.m_TargetHeight; in CreateResizeBilinear()
521 std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateRsqrt(const RsqrtQueueDescriptor &descriptor, in CreateRsqrt() argument
524 IgnoreUnused(descriptor); in CreateRsqrt()
532 std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateSlice(const SliceQueueDescriptor& descriptor, in CreateSlice() argument
535 return std::make_unique<NeonSliceWorkload>(descriptor, info); in CreateSlice()
538 …:unique_ptr<IWorkload> NeonWorkloadFactory::CreateSoftmax(const SoftmaxQueueDescriptor& descriptor, in CreateSoftmax() argument
541 …return std::make_unique<NeonSoftmaxWorkload>(descriptor, info, m_MemoryManager->GetIntraLayerManag… in CreateSoftmax()
544 …orkload> NeonWorkloadFactory::CreateSpaceToBatchNd(const SpaceToBatchNdQueueDescriptor& descriptor, in CreateSpaceToBatchNd() argument
547 return std::make_unique<NeonSpaceToBatchNdWorkload>(descriptor, info); in CreateSpaceToBatchNd()
550 …r<IWorkload> NeonWorkloadFactory::CreateSpaceToDepth(const SpaceToDepthQueueDescriptor& descriptor, in CreateSpaceToDepth() argument
553 return std::make_unique<NeonSpaceToDepthWorkload>(descriptor, info); in CreateSpaceToDepth()
556 …nique_ptr<IWorkload> NeonWorkloadFactory::CreateSplitter(const SplitterQueueDescriptor& descriptor, in CreateSplitter() argument
559 return std::make_unique<NeonSplitterWorkload>(descriptor, info); in CreateSplitter()
562 std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateStack(const StackQueueDescriptor& descriptor, in CreateStack() argument
565 return std::make_unique<NeonStackWorkload>(descriptor, info); in CreateStack()
568 …r<IWorkload> NeonWorkloadFactory::CreateStridedSlice(const StridedSliceQueueDescriptor& descriptor, in CreateStridedSlice() argument
571 return std::make_unique<NeonStridedSliceWorkload>(descriptor, info); in CreateStridedSlice()
575 const SubtractionQueueDescriptor& descriptor, const WorkloadInfo& info) const in CreateSubtraction() argument
577 return std::make_unique<NeonSubtractionWorkload>(descriptor, info); in CreateSubtraction()
580 …<armnn::IWorkload> NeonWorkloadFactory::CreateTranspose(const TransposeQueueDescriptor& descriptor, in CreateTranspose() argument
583 return std::make_unique<NeonTransposeWorkload>(descriptor, info); in CreateTranspose()
587 const TransposeConvolution2dQueueDescriptor &descriptor, in CreateTransposeConvolution2d() argument
590 return std::make_unique<NeonTransposeConvolution2dWorkload>(descriptor, info, in CreateTransposeConvolution2d()