Lines Matching full:info
26 const WorkloadInfo& info) const in MakeWorkload()
29 (descriptor, info); in MakeWorkload()
33 bool IsDataType(const WorkloadInfo& info) in IsDataType() argument
36 …auto it = std::find_if(std::begin(info.m_InputTensorInfos), std::end(info.m_InputTensorInfos), che… in IsDataType()
37 if (it != std::end(info.m_InputTensorInfos)) in IsDataType()
41 …it = std::find_if(std::begin(info.m_OutputTensorInfos), std::end(info.m_OutputTensorInfos), checkT… in IsDataType()
42 if (it != std::end(info.m_OutputTensorInfos)) in IsDataType()
49 bool IsSigned32(const WorkloadInfo& info) in IsSigned32() argument
51 return IsDataType<DataType::Signed32>(info); in IsSigned32()
54 bool IsBFloat16(const WorkloadInfo& info) in IsBFloat16() argument
56 return IsDataType<DataType::BFloat16>(info); in IsBFloat16()
59 bool IsFloat16(const WorkloadInfo& info) in IsFloat16() argument
61 return IsDataType<DataType::Float16>(info); in IsFloat16()
64 bool IsQSymmS16(const WorkloadInfo& info) in IsQSymmS16() argument
66 return IsDataType<DataType::QSymmS16>(info); in IsQSymmS16()
69 bool IsQSymmS8(const WorkloadInfo& info) in IsQSymmS8() argument
71 return IsDataType<DataType::QSymmS8>(info); in IsQSymmS8()
74 bool IsQAsymmS8(const WorkloadInfo& info) in IsQAsymmS8() argument
76 return IsDataType<DataType::QAsymmS8>(info); in IsQAsymmS8()
79 bool IsQAsymmU8(const WorkloadInfo& info) in IsQAsymmU8() argument
81 return IsDataType<DataType::QAsymmU8>(info); in IsQAsymmU8()
134 const WorkloadInfo& info) const in CreateAbs()
140 return CreateElementwiseUnary(elementwiseUnaryDescriptor, info); in CreateAbs()
144 const WorkloadInfo& info) const in CreateActivation()
146 return std::make_unique<RefActivationWorkload>(descriptor, info); in CreateActivation()
150 const WorkloadInfo& info) const in CreateAddition()
152 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateAddition()
154 return std::make_unique<RefAdditionWorkload<int32_t>>(descriptor, info); in CreateAddition()
158 return std::make_unique<RefAdditionWorkload<float>>(descriptor, info); in CreateAddition()
163 const WorkloadInfo& info) const in CreateArgMinMax()
165 return std::make_unique<RefArgMinMaxWorkload>(descriptor, info); in CreateArgMinMax()
170 const WorkloadInfo& info) const in CreateBatchNormalization()
172 return std::make_unique<RefBatchNormalizationWorkload>(descriptor, info); in CreateBatchNormalization()
176 const WorkloadInfo& info) const in CreateBatchToSpaceNd()
178 return std::make_unique<RefBatchToSpaceNdWorkload>(descriptor, info); in CreateBatchToSpaceNd()
182 const WorkloadInfo& info) const in CreateComparison()
184 return std::make_unique<RefComparisonWorkload>(descriptor, info); in CreateComparison()
188 const WorkloadInfo& info) const in CreateConcat()
190 return std::make_unique<RefConcatWorkload>(descriptor, info); in CreateConcat()
194 const WorkloadInfo& info) const in CreateConstant()
196 return std::make_unique<RefConstantWorkload>(descriptor, info); in CreateConstant()
201 const WorkloadInfo& info) const in CreateConvertBf16ToFp32()
203 return std::make_unique<RefConvertBf16ToFp32Workload>(descriptor, info); in CreateConvertBf16ToFp32()
208 const WorkloadInfo& info) const in CreateConvertFp16ToFp32()
210 return std::make_unique<RefConvertFp16ToFp32Workload>(descriptor, info); in CreateConvertFp16ToFp32()
215 const WorkloadInfo& info) const in CreateConvertFp32ToBf16()
217 return std::make_unique<RefConvertFp32ToBf16Workload>(descriptor, info); in CreateConvertFp32ToBf16()
222 const WorkloadInfo& info) const in CreateConvertFp32ToFp16()
224 return std::make_unique<RefConvertFp32ToFp16Workload>(descriptor, info); in CreateConvertFp32ToFp16()
228 const WorkloadInfo& info) const in CreateConvolution2d()
230 return std::make_unique<RefConvolution2dWorkload>(descriptor, info); in CreateConvolution2d()
234 const WorkloadInfo& info) const in CreateDebug()
236 if (IsBFloat16(info)) in CreateDebug()
238 return std::make_unique<RefDebugBFloat16Workload>(descriptor, info); in CreateDebug()
240 if (IsFloat16(info)) in CreateDebug()
242 return std::make_unique<RefDebugFloat16Workload>(descriptor, info); in CreateDebug()
244 if (IsQSymmS16(info)) in CreateDebug()
246 return std::make_unique<RefDebugQSymmS16Workload>(descriptor, info); in CreateDebug()
248 if (IsQSymmS8(info)) in CreateDebug()
250 return std::make_unique<RefDebugQSymmS8Workload>(descriptor, info); in CreateDebug()
252 if (IsQAsymmU8(info)) in CreateDebug()
254 return std::make_unique<RefDebugQAsymmU8Workload>(descriptor, info); in CreateDebug()
256 if (IsQAsymmS8(info)) in CreateDebug()
258 return std::make_unique<RefDebugQAsymmS8Workload>(descriptor, info); in CreateDebug()
260 if (IsSigned32(info)) in CreateDebug()
262 return std::make_unique<RefDebugSigned32Workload>(descriptor, info); in CreateDebug()
265 return MakeWorkload<RefDebugFloat32Workload, RefDebugQAsymmU8Workload>(descriptor, info); in CreateDebug()
269 const WorkloadInfo& info) const in CreateDepthToSpace()
271 return std::make_unique<RefDepthToSpaceWorkload>(descriptor, info); in CreateDepthToSpace()
276 const WorkloadInfo& info) const in CreateDepthwiseConvolution2d()
278 return std::make_unique<RefDepthwiseConvolution2dWorkload>(descriptor, info); in CreateDepthwiseConvolution2d()
282 const WorkloadInfo& info) const in CreateDequantize()
284 return std::make_unique<RefDequantizeWorkload>(descriptor, info); in CreateDequantize()
289 const WorkloadInfo& info) const in CreateDetectionPostProcess()
291 return std::make_unique<RefDetectionPostProcessWorkload>(descriptor, info); in CreateDetectionPostProcess()
295 const WorkloadInfo& info) const in CreateDivision()
297 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateDivision()
299 return std::make_unique<RefDivisionWorkload<int32_t>>(descriptor, info); in CreateDivision()
303 return std::make_unique<RefDivisionWorkload<float>>(descriptor, info); in CreateDivision()
308 … const WorkloadInfo& info) const in CreateElementwiseUnary()
312 return std::make_unique<RefLogicalUnaryWorkload>(descriptor, info); in CreateElementwiseUnary()
314 return std::make_unique<RefElementwiseUnaryWorkload>(descriptor, info); in CreateElementwiseUnary()
318 const WorkloadInfo& info) const in CreateEqual()
324 return CreateComparison(comparisonDescriptor, info); in CreateEqual()
328 … const WorkloadInfo& info) const in CreateFakeQuantization()
330 return MakeWorkload<RefFakeQuantizationFloat32Workload, NullWorkload>(descriptor, info); in CreateFakeQuantization()
334 const WorkloadInfo& info) const in CreateFill()
336 return std::make_unique<RefFillWorkload>(descriptor, info); in CreateFill()
340 const WorkloadInfo& info) const in CreateFloor()
342 if(IsQuantizedType(info.m_InputTensorInfos[0].GetDataType())) in CreateFloor()
348 return std::make_unique<RefFloorWorkload>(descriptor, info); in CreateFloor()
354 const WorkloadInfo& info) const in CreateFullyConnected()
356 return std::make_unique<RefFullyConnectedWorkload>(descriptor, info); in CreateFullyConnected()
360 const WorkloadInfo& info) const in CreateGather()
362 return std::make_unique<RefGatherWorkload>(descriptor, info); in CreateGather()
366 const WorkloadInfo& info) const in CreateGreater()
372 return CreateComparison(comparisonDescriptor, info); in CreateGreater()
376 const WorkloadInfo& info) const in CreateInput()
378 if (info.m_InputTensorInfos.empty() ) in CreateInput()
382 if (info.m_OutputTensorInfos.empty()) in CreateInput()
387 if (info.m_InputTensorInfos[0].GetNumBytes() != info.m_OutputTensorInfos[0].GetNumBytes()) in CreateInput()
392 return std::make_unique<CopyMemGenericWorkload>(descriptor, info); in CreateInput()
397 const WorkloadInfo& info) const in CreateInstanceNormalization()
399 return std::make_unique<RefInstanceNormalizationWorkload>(descriptor, info); in CreateInstanceNormalization()
403 const WorkloadInfo& info) const in CreateL2Normalization()
405 return std::make_unique<RefL2NormalizationWorkload>(descriptor, info); in CreateL2Normalization()
409 const WorkloadInfo& info) const in CreateLogicalBinary()
411 return std::make_unique<RefLogicalBinaryWorkload>(descriptor, info); in CreateLogicalBinary()
415 const WorkloadInfo& info) const in CreateLogSoftmax()
417 return std::make_unique<RefLogSoftmaxWorkload>(descriptor, info); in CreateLogSoftmax()
421 const WorkloadInfo& info) const in CreateLstm()
423 return std::make_unique<RefLstmWorkload>(descriptor, info); in CreateLstm()
427 const WorkloadInfo& info) const in CreateMaximum()
429 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateMaximum()
431 return std::make_unique<RefMaximumWorkload<int32_t>>(descriptor, info); in CreateMaximum()
435 return std::make_unique<RefMaximumWorkload<float>>(descriptor, info); in CreateMaximum()
440 const WorkloadInfo& info) const in CreateMean()
442 return std::make_unique<RefMeanWorkload>(descriptor, info); in CreateMean()
446 const WorkloadInfo& info) const in CreateMemCopy()
452 return std::make_unique<CopyMemGenericWorkload>(descriptor, info); in CreateMemCopy()
456 const WorkloadInfo& info) const in CreateMemImport()
462 return std::make_unique<ImportMemGenericWorkload>(descriptor, info); in CreateMemImport()
466 const WorkloadInfo& info) const in CreateMerger()
468 return CreateConcat(descriptor, info); in CreateMerger()
472 const WorkloadInfo& info) const in CreateMinimum()
474 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateMinimum()
476 return std::make_unique<RefMinimumWorkload<int32_t>>(descriptor, info); in CreateMinimum()
480 return std::make_unique<RefMinimumWorkload<float>>(descriptor, info); in CreateMinimum()
485 const WorkloadInfo& info) const in CreateMultiplication()
487 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateMultiplication()
489 return std::make_unique<RefMultiplicationWorkload<int32_t>>(descriptor, info); in CreateMultiplication()
493 return std::make_unique<RefMultiplicationWorkload<float>>(descriptor, info); in CreateMultiplication()
498 const WorkloadInfo& info) const in CreateNormalization()
500 return std::make_unique<RefNormalizationWorkload>(descriptor, info); in CreateNormalization()
504 const WorkloadInfo& info) const in CreateOutput()
506 if (info.m_InputTensorInfos.empty() ) in CreateOutput()
510 if (info.m_OutputTensorInfos.empty()) in CreateOutput()
514 if (info.m_InputTensorInfos[0].GetNumBytes() != info.m_OutputTensorInfos[0].GetNumBytes()) in CreateOutput()
519 return std::make_unique<CopyMemGenericWorkload>(descriptor, info); in CreateOutput()
523 const WorkloadInfo& info) const in CreatePad()
525 return std::make_unique<RefPadWorkload>(descriptor, info); in CreatePad()
529 const WorkloadInfo& info) const in CreatePermute()
531 if (IsQSymmS16(info)) in CreatePermute()
533 return std::make_unique<RefPermuteQSymm16Workload>(descriptor, info); in CreatePermute()
535 else if (IsBFloat16(info)) in CreatePermute()
537 return std::make_unique<RefPermuteBFloat16Workload>(descriptor, info); in CreatePermute()
539 else if (IsQAsymmS8(info)) in CreatePermute()
541 return std::make_unique<RefPermuteQAsymmS8Workload>(descriptor, info); in CreatePermute()
544 NullWorkload, NullWorkload, NullWorkload>(descriptor, info); in CreatePermute()
548 const WorkloadInfo& info) const in CreatePooling2d()
550 return std::make_unique<RefPooling2dWorkload>(descriptor, info); in CreatePooling2d()
554 const WorkloadInfo& /*info*/) const in CreatePreCompiled()
560 const WorkloadInfo& info) const in CreatePrelu()
562 return std::make_unique<RefPreluWorkload>(descriptor, info); in CreatePrelu()
566 const WorkloadInfo& info) const in CreateQLstm()
568 return std::make_unique<RefQLstmWorkload>(descriptor, info); in CreateQLstm()
572 const WorkloadInfo& info) const in CreateQuantize()
574 return std::make_unique<RefQuantizeWorkload>(descriptor, info); in CreateQuantize()
578 const WorkloadInfo& info) const in CreateRank()
580 return std::make_unique<RefRankWorkload>(descriptor, info); in CreateRank()
584 const WorkloadInfo& info) const in CreateReshape()
586 return std::make_unique<RefReshapeWorkload>(descriptor, info); in CreateReshape()
590 const WorkloadInfo& info) const in CreateResize()
592 return std::make_unique<RefResizeWorkload>(descriptor, info); in CreateResize()
596 const WorkloadInfo& info) const in CreateResizeBilinear()
604 return CreateResize(resizeDescriptor, info); in CreateResizeBilinear()
608 const WorkloadInfo& info) const in CreateRsqrt()
614 return CreateElementwiseUnary(elementwiseUnaryDescriptor, info); in CreateRsqrt()
618 const WorkloadInfo& info) const in CreateSlice()
620 return std::make_unique<RefSliceWorkload>(descriptor, info); in CreateSlice()
624 const WorkloadInfo& info) const in CreateSoftmax()
626 return std::make_unique<RefSoftmaxWorkload>(descriptor, info); in CreateSoftmax()
630 const WorkloadInfo& info) const in CreateSpaceToBatchNd()
632 return std::make_unique<RefSpaceToBatchNdWorkload>(descriptor, info); in CreateSpaceToBatchNd()
636 const WorkloadInfo& info) const in CreateSpaceToDepth()
638 return std::make_unique<RefSpaceToDepthWorkload>(descriptor, info); in CreateSpaceToDepth()
642 const WorkloadInfo& info) const in CreateSplitter()
644 return std::make_unique<RefSplitterWorkload>(descriptor, info); in CreateSplitter()
648 const WorkloadInfo& info) const in CreateStack()
650 return std::make_unique<RefStackWorkload>(descriptor, info); in CreateStack()
654 const WorkloadInfo& info) const in CreateStridedSlice()
656 return std::make_unique<RefStridedSliceWorkload>(descriptor, info); in CreateStridedSlice()
660 const WorkloadInfo& info) const in CreateSubtraction()
662 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateSubtraction()
664 return std::make_unique<RefSubtractionWorkload<int32_t>>(descriptor, info); in CreateSubtraction()
668 return std::make_unique<RefSubtractionWorkload<float>>(descriptor, info); in CreateSubtraction()
673 const WorkloadInfo& info) const in CreateTranspose()
675 if (IsQSymmS16(info)) in CreateTranspose()
677 return std::make_unique<RefTransposeQSymm16Workload>(descriptor, info); in CreateTranspose()
679 else if (IsBFloat16(info)) in CreateTranspose()
681 return std::make_unique<RefTransposeBFloat16Workload>(descriptor, info); in CreateTranspose()
683 else if (IsQAsymmS8(info)) in CreateTranspose()
685 return std::make_unique<RefTransposeQAsymmS8Workload>(descriptor, info); in CreateTranspose()
688 NullWorkload, NullWorkload, NullWorkload>(descriptor, info); in CreateTranspose()
693 const WorkloadInfo& info) const in CreateTransposeConvolution2d()
695 return std::make_unique<RefTransposeConvolution2dWorkload>(descriptor, info); in CreateTransposeConvolution2d()