Lines Matching full:info
25 const WorkloadInfo& info) const in MakeWorkload()
28 (descriptor, info); in MakeWorkload()
32 bool IsDataType(const WorkloadInfo& info) in IsDataType() argument
35 …auto it = std::find_if(std::begin(info.m_InputTensorInfos), std::end(info.m_InputTensorInfos), che… in IsDataType()
36 if (it != std::end(info.m_InputTensorInfos)) in IsDataType()
40 …it = std::find_if(std::begin(info.m_OutputTensorInfos), std::end(info.m_OutputTensorInfos), checkT… in IsDataType()
41 if (it != std::end(info.m_OutputTensorInfos)) in IsDataType()
48 bool IsSigned32(const WorkloadInfo& info) in IsSigned32() argument
50 return IsDataType<DataType::Signed32>(info); in IsSigned32()
53 bool IsBFloat16(const WorkloadInfo& info) in IsBFloat16() argument
55 return IsDataType<DataType::BFloat16>(info); in IsBFloat16()
58 bool IsFloat16(const WorkloadInfo& info) in IsFloat16() argument
60 return IsDataType<DataType::Float16>(info); in IsFloat16()
63 bool IsQSymmS16(const WorkloadInfo& info) in IsQSymmS16() argument
65 return IsDataType<DataType::QSymmS16>(info); in IsQSymmS16()
68 bool IsQSymmS8(const WorkloadInfo& info) in IsQSymmS8() argument
70 return IsDataType<DataType::QSymmS8>(info); in IsQSymmS8()
73 bool IsQAsymmS8(const WorkloadInfo& info) in IsQAsymmS8() argument
75 return IsDataType<DataType::QAsymmS8>(info); in IsQAsymmS8()
78 bool IsQAsymmU8(const WorkloadInfo& info) in IsQAsymmU8() argument
80 return IsDataType<DataType::QAsymmU8>(info); in IsQAsymmU8()
146 const WorkloadInfo& info) const in CreateWorkload()
153 return std::make_unique<RefActivationWorkload>(*activationQueueDescriptor, info); in CreateWorkload()
159 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateWorkload()
161 … return std::make_unique<RefAdditionWorkload<int32_t>>(*additionQueueDescriptor, info); in CreateWorkload()
165 return std::make_unique<RefAdditionWorkload<float>>(*additionQueueDescriptor, info); in CreateWorkload()
171 return std::make_unique<RefArgMinMaxWorkload>(*argMinMaxQueueDescriptor, info); in CreateWorkload()
176 return std::make_unique<RefBatchMatMulWorkload>(*batchMatMulQueueDescriptor, info); in CreateWorkload()
181 return std::make_unique<RefBatchNormalizationWorkload>(*batchNormQueueDescriptor, info); in CreateWorkload()
187 … return std::make_unique<RefBatchToSpaceNdWorkload>(*batchToSpaceNdQueueDescriptor, info); in CreateWorkload()
192 return std::make_unique<RefCastWorkload>(*castQueueDescriptor, info); in CreateWorkload()
198 … return std::make_unique<RefChannelShuffleWorkload>(*channelShuffleQueueDescriptor, info); in CreateWorkload()
203 return std::make_unique<RefComparisonWorkload>(*comparisonQueueDescriptor, info); in CreateWorkload()
208 return std::make_unique<RefConcatWorkload>(*concatQueueDescriptor, info); in CreateWorkload()
213 return std::make_unique<RefConstantWorkload>(*constantQueueDescriptor, info); in CreateWorkload()
219 … return std::make_unique<RefConvertFp16ToFp32Workload>(*convertFp16ToFp32QueueDescriptor, info); in CreateWorkload()
225 … return std::make_unique<RefConvertFp32ToFp16Workload>(*convertFp32ToFp16QueueDescriptor, info); in CreateWorkload()
230 return std::make_unique<RefConvolution2dWorkload>(*convolution2dQueueDescriptor, info); in CreateWorkload()
235 return std::make_unique<RefConvolution3dWorkload>(*convolution3dQueueDescriptor, info); in CreateWorkload()
240 if (IsBFloat16(info)) in CreateWorkload()
242 return std::make_unique<RefDebugBFloat16Workload>(*debugQueueDescriptor, info); in CreateWorkload()
244 if (IsFloat16(info)) in CreateWorkload()
246 return std::make_unique<RefDebugFloat16Workload>(*debugQueueDescriptor, info); in CreateWorkload()
248 if (IsQSymmS16(info)) in CreateWorkload()
250 return std::make_unique<RefDebugQSymmS16Workload>(*debugQueueDescriptor, info); in CreateWorkload()
252 if (IsQSymmS8(info)) in CreateWorkload()
254 return std::make_unique<RefDebugQSymmS8Workload>(*debugQueueDescriptor, info); in CreateWorkload()
256 if (IsQAsymmU8(info)) in CreateWorkload()
258 return std::make_unique<RefDebugQAsymmU8Workload>(*debugQueueDescriptor, info); in CreateWorkload()
260 if (IsQAsymmS8(info)) in CreateWorkload()
262 return std::make_unique<RefDebugQAsymmS8Workload>(*debugQueueDescriptor, info); in CreateWorkload()
264 if (IsSigned32(info)) in CreateWorkload()
266 return std::make_unique<RefDebugSigned32Workload>(*debugQueueDescriptor, info); in CreateWorkload()
269 …eturn MakeWorkload<RefDebugFloat32Workload, RefDebugQAsymmU8Workload>(*debugQueueDescriptor, info); in CreateWorkload()
274 return std::make_unique<RefDepthToSpaceWorkload>(*depthToSpaceQueueDescriptor, info); in CreateWorkload()
280 … std::make_unique<RefDepthwiseConvolution2dWorkload>(*depthwiseConvolution2DQueueDescriptor, info); in CreateWorkload()
285 return std::make_unique<RefDequantizeWorkload>(*dequantizeQueueDescriptor, info); in CreateWorkload()
291 …turn std::make_unique<RefDetectionPostProcessWorkload>(*detectionPostProcessQueueDescriptor, info); in CreateWorkload()
296 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateWorkload()
298 … return std::make_unique<RefDivisionWorkload<int32_t>>(*divisionQueueDescriptor, info); in CreateWorkload()
302 return std::make_unique<RefDivisionWorkload<float>>(*divisionQueueDescriptor, info); in CreateWorkload()
309 … return std::make_unique<RefElementwiseBinaryWorkload>(*elementwiseBinaryQueueDescriptor, info); in CreateWorkload()
317 … return std::make_unique<RefLogicalUnaryWorkload>(*elementwiseUnaryQueueDescriptor, info); in CreateWorkload()
319 … return std::make_unique<RefElementwiseUnaryWorkload>(*elementwiseUnaryQueueDescriptor, info); in CreateWorkload()
325 …eturn std::make_unique<RefFakeQuantizationFloat32Workload>(*fakeQuantizationQueueDescriptor, info); in CreateWorkload()
330 return std::make_unique<RefFillWorkload>(*fillQueueDescriptor, info); in CreateWorkload()
335 if(IsQuantizedType(info.m_InputTensorInfos[0].GetDataType())) in CreateWorkload()
341 return std::make_unique<RefFloorWorkload>(*floorQueueDescriptor, info); in CreateWorkload()
348 … return std::make_unique<RefFullyConnectedWorkload>(*fullyConnectedQueueDescriptor, info); in CreateWorkload()
353 return std::make_unique<RefGatherWorkload>(*gatherQueueDescriptor, info); in CreateWorkload()
358 return std::make_unique<RefGatherNdWorkload>(*gatherNdQueueDescriptor, info); in CreateWorkload()
363 if (info.m_InputTensorInfos.empty() ) in CreateWorkload()
367 if (info.m_OutputTensorInfos.empty()) in CreateWorkload()
372 … if (info.m_InputTensorInfos[0].GetNumBytes() != info.m_OutputTensorInfos[0].GetNumBytes()) in CreateWorkload()
378 return std::make_unique<CopyMemGenericWorkload>(*inputQueueDescriptor, info); in CreateWorkload()
384 …rn std::make_unique<RefInstanceNormalizationWorkload>(*instanceNormalizationQueueDescriptor, info); in CreateWorkload()
390 … return std::make_unique<RefL2NormalizationWorkload>(*l2NormalizationQueueDescriptor, info); in CreateWorkload()
395 return std::make_unique<RefLogicalBinaryWorkload>(*logicalBinaryQueueDescriptor, info); in CreateWorkload()
400 return std::make_unique<RefLogSoftmaxWorkload>(*logSoftmaxQueueDescriptor, info); in CreateWorkload()
405 return std::make_unique<RefLstmWorkload>(*lstmQueueDescriptor, info); in CreateWorkload()
410 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateWorkload()
412 return std::make_unique<RefMaximumWorkload<int32_t>>(*maximumQueueDescriptor, info); in CreateWorkload()
416 return std::make_unique<RefMaximumWorkload<float>>(*maximumQueueDescriptor, info); in CreateWorkload()
422 return std::make_unique<RefMeanWorkload>(*meanQueueDescriptor, info); in CreateWorkload()
431 return std::make_unique<CopyMemGenericWorkload>(*memCopyQueueDescriptor, info); in CreateWorkload()
440 return std::make_unique<ImportMemGenericWorkload>(*memImportQueueDescriptor, info); in CreateWorkload()
445 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateWorkload()
447 return std::make_unique<RefMinimumWorkload<int32_t>>(*minimumQueueDescriptor, info); in CreateWorkload()
451 return std::make_unique<RefMinimumWorkload<float>>(*minimumQueueDescriptor, info); in CreateWorkload()
458 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateWorkload()
460 … return std::make_unique<RefMultiplicationWorkload<int32_t>>(*multiplicationQueueDescriptor, info); in CreateWorkload()
464 … return std::make_unique<RefMultiplicationWorkload<float>>(*multiplicationQueueDescriptor, info); in CreateWorkload()
470 return std::make_unique<RefNormalizationWorkload>(*normalizationQueueDescriptor, info); in CreateWorkload()
475 if (info.m_InputTensorInfos.empty() ) in CreateWorkload()
479 if (info.m_OutputTensorInfos.empty()) in CreateWorkload()
483 … if (info.m_InputTensorInfos[0].GetNumBytes() != info.m_OutputTensorInfos[0].GetNumBytes()) in CreateWorkload()
489 return std::make_unique<CopyMemGenericWorkload>(*outputQueueDescriptor, info); in CreateWorkload()
494 return std::make_unique<RefPadWorkload>(*padQueueDescriptor, info); in CreateWorkload()
499 if (IsQSymmS16(info)) in CreateWorkload()
501 return std::make_unique<RefPermuteQSymm16Workload>(*permuteQueueDescriptor, info); in CreateWorkload()
503 else if (IsBFloat16(info)) in CreateWorkload()
505 return std::make_unique<RefPermuteBFloat16Workload>(*permuteQueueDescriptor, info); in CreateWorkload()
507 else if (IsQAsymmS8(info)) in CreateWorkload()
509 return std::make_unique<RefPermuteQAsymmS8Workload>(*permuteQueueDescriptor, info); in CreateWorkload()
512 NullWorkload, NullWorkload, NullWorkload>(*permuteQueueDescriptor, info); in CreateWorkload()
517 return std::make_unique<RefPooling2dWorkload>(*pooling2dQueueDescriptor, info); in CreateWorkload()
522 return std::make_unique<RefPooling3dWorkload>(*pooling3dQueueDescriptor, info); in CreateWorkload()
531 return std::make_unique<RefPreluWorkload>(*preluQueueDescriptor, info); in CreateWorkload()
536 return std::make_unique<RefQLstmWorkload>(*qlstmQueueDescriptor, info); in CreateWorkload()
541 return std::make_unique<RefQuantizeWorkload>(*quantizeQueueDescriptor, info); in CreateWorkload()
546 return std::make_unique<RefRankWorkload>(*rankQueueDescriptor, info); in CreateWorkload()
551 return std::make_unique<RefReduceWorkload>(*reduceQueueDescriptor, info); in CreateWorkload()
556 return std::make_unique<RefReshapeWorkload>(*reshapeQueueDescriptor, info); in CreateWorkload()
561 return std::make_unique<RefResizeWorkload>(*resizeQueueDescriptor, info); in CreateWorkload()
566 return std::make_unique<RefShapeWorkload>(*shapeQueueDescriptor, info); in CreateWorkload()
571 return std::make_unique<RefSliceWorkload>(*sliceQueueDescriptor, info); in CreateWorkload()
576 return std::make_unique<RefSoftmaxWorkload>(*softmaxQueueDescriptor, info); in CreateWorkload()
582 … return std::make_unique<RefSpaceToBatchNdWorkload>(*spaceToBatchNdQueueDescriptor, info); in CreateWorkload()
587 return std::make_unique<RefSpaceToDepthWorkload>(*spaceToDepthQueueDescriptor, info); in CreateWorkload()
592 return std::make_unique<RefSplitterWorkload>(*splitterQueueDescriptor, info); in CreateWorkload()
597 return std::make_unique<RefStackWorkload>(*stackQueueDescriptor, info); in CreateWorkload()
602 return std::make_unique<RefStridedSliceWorkload>(*stridedSliceQueueDescriptor, info); in CreateWorkload()
607 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateWorkload()
609 … return std::make_unique<RefSubtractionWorkload<int32_t>>(*subtractionQueueDescriptor, info); in CreateWorkload()
613 … return std::make_unique<RefSubtractionWorkload<float>>(*subtractionQueueDescriptor, info); in CreateWorkload()
619 if (IsQSymmS16(info)) in CreateWorkload()
621 … return std::make_unique<RefTransposeQSymm16Workload>(*transposeQueueDescriptor, info); in CreateWorkload()
623 else if (IsBFloat16(info)) in CreateWorkload()
625 … return std::make_unique<RefTransposeBFloat16Workload>(*transposeQueueDescriptor, info); in CreateWorkload()
627 else if (IsQAsymmS8(info)) in CreateWorkload()
629 … return std::make_unique<RefTransposeQAsymmS8Workload>(*transposeQueueDescriptor, info); in CreateWorkload()
633 (*transposeQueueDescriptor, info); in CreateWorkload()
639 … std::make_unique<RefTransposeConvolution2dWorkload>(*transposeConvolution2dQueueDescriptor, info); in CreateWorkload()
646 info); in CreateWorkload()
654 const WorkloadInfo& info) const in CreateActivation()
656 return std::make_unique<RefActivationWorkload>(descriptor, info); in CreateActivation()
660 const WorkloadInfo& info) const in CreateAddition()
662 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateAddition()
664 return std::make_unique<RefAdditionWorkload<int32_t>>(descriptor, info); in CreateAddition()
668 return std::make_unique<RefAdditionWorkload<float>>(descriptor, info); in CreateAddition()
673 const WorkloadInfo& info) const in CreateArgMinMax()
675 return std::make_unique<RefArgMinMaxWorkload>(descriptor, info); in CreateArgMinMax()
680 const WorkloadInfo& info) const in CreateBatchNormalization()
682 return std::make_unique<RefBatchNormalizationWorkload>(descriptor, info); in CreateBatchNormalization()
686 const WorkloadInfo& info) const in CreateBatchToSpaceNd()
688 return std::make_unique<RefBatchToSpaceNdWorkload>(descriptor, info); in CreateBatchToSpaceNd()
692 const WorkloadInfo& info) const in CreateCast()
694 return std::make_unique<RefCastWorkload>(descriptor, info); in CreateCast()
698 const WorkloadInfo &info) const in CreateChannelShuffle()
700 return std::make_unique<RefChannelShuffleWorkload>(descriptor,info); in CreateChannelShuffle()
704 const WorkloadInfo& info) const in CreateComparison()
706 return std::make_unique<RefComparisonWorkload>(descriptor, info); in CreateComparison()
710 const WorkloadInfo& info) const in CreateConcat()
712 return std::make_unique<RefConcatWorkload>(descriptor, info); in CreateConcat()
716 const WorkloadInfo& info) const in CreateConstant()
718 return std::make_unique<RefConstantWorkload>(descriptor, info); in CreateConstant()
723 const WorkloadInfo& info) const in CreateConvertFp16ToFp32()
725 return std::make_unique<RefConvertFp16ToFp32Workload>(descriptor, info); in CreateConvertFp16ToFp32()
730 const WorkloadInfo& info) const in CreateConvertFp32ToFp16()
732 return std::make_unique<RefConvertFp32ToFp16Workload>(descriptor, info); in CreateConvertFp32ToFp16()
736 const WorkloadInfo& info) const in CreateConvolution2d()
738 return std::make_unique<RefConvolution2dWorkload>(descriptor, info); in CreateConvolution2d()
742 const WorkloadInfo& info) const in CreateConvolution3d()
744 return std::make_unique<RefConvolution3dWorkload>(descriptor, info); in CreateConvolution3d()
748 const WorkloadInfo& info) const in CreateDebug()
750 if (IsBFloat16(info)) in CreateDebug()
752 return std::make_unique<RefDebugBFloat16Workload>(descriptor, info); in CreateDebug()
754 if (IsFloat16(info)) in CreateDebug()
756 return std::make_unique<RefDebugFloat16Workload>(descriptor, info); in CreateDebug()
758 if (IsQSymmS16(info)) in CreateDebug()
760 return std::make_unique<RefDebugQSymmS16Workload>(descriptor, info); in CreateDebug()
762 if (IsQSymmS8(info)) in CreateDebug()
764 return std::make_unique<RefDebugQSymmS8Workload>(descriptor, info); in CreateDebug()
766 if (IsQAsymmU8(info)) in CreateDebug()
768 return std::make_unique<RefDebugQAsymmU8Workload>(descriptor, info); in CreateDebug()
770 if (IsQAsymmS8(info)) in CreateDebug()
772 return std::make_unique<RefDebugQAsymmS8Workload>(descriptor, info); in CreateDebug()
774 if (IsSigned32(info)) in CreateDebug()
776 return std::make_unique<RefDebugSigned32Workload>(descriptor, info); in CreateDebug()
779 return MakeWorkload<RefDebugFloat32Workload, RefDebugQAsymmU8Workload>(descriptor, info); in CreateDebug()
783 const WorkloadInfo& info) const in CreateDepthToSpace()
785 return std::make_unique<RefDepthToSpaceWorkload>(descriptor, info); in CreateDepthToSpace()
790 const WorkloadInfo& info) const in CreateDepthwiseConvolution2d()
792 return std::make_unique<RefDepthwiseConvolution2dWorkload>(descriptor, info); in CreateDepthwiseConvolution2d()
796 const WorkloadInfo& info) const in CreateDequantize()
798 return std::make_unique<RefDequantizeWorkload>(descriptor, info); in CreateDequantize()
803 const WorkloadInfo& info) const in CreateDetectionPostProcess()
805 return std::make_unique<RefDetectionPostProcessWorkload>(descriptor, info); in CreateDetectionPostProcess()
809 const WorkloadInfo& info) const in CreateDivision()
811 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateDivision()
813 return std::make_unique<RefDivisionWorkload<int32_t>>(descriptor, info); in CreateDivision()
817 return std::make_unique<RefDivisionWorkload<float>>(descriptor, info); in CreateDivision()
822 … const WorkloadInfo& info) const in CreateElementwiseUnary()
826 return std::make_unique<RefLogicalUnaryWorkload>(descriptor, info); in CreateElementwiseUnary()
828 return std::make_unique<RefElementwiseUnaryWorkload>(descriptor, info); in CreateElementwiseUnary()
832 … const WorkloadInfo& info) const in CreateFakeQuantization()
834 return MakeWorkload<RefFakeQuantizationFloat32Workload, NullWorkload>(descriptor, info); in CreateFakeQuantization()
838 const WorkloadInfo& info) const in CreateFill()
840 return std::make_unique<RefFillWorkload>(descriptor, info); in CreateFill()
844 const WorkloadInfo& info) const in CreateFloor()
846 if(IsQuantizedType(info.m_InputTensorInfos[0].GetDataType())) in CreateFloor()
852 return std::make_unique<RefFloorWorkload>(descriptor, info); in CreateFloor()
858 const WorkloadInfo& info) const in CreateFullyConnected()
860 return std::make_unique<RefFullyConnectedWorkload>(descriptor, info); in CreateFullyConnected()
864 const WorkloadInfo& info) const in CreateGather()
866 return std::make_unique<RefGatherWorkload>(descriptor, info); in CreateGather()
870 const WorkloadInfo& info) const in CreateInput()
872 if (info.m_InputTensorInfos.empty() ) in CreateInput()
876 if (info.m_OutputTensorInfos.empty()) in CreateInput()
881 if (info.m_InputTensorInfos[0].GetNumBytes() != info.m_OutputTensorInfos[0].GetNumBytes()) in CreateInput()
886 return std::make_unique<CopyMemGenericWorkload>(descriptor, info); in CreateInput()
891 const WorkloadInfo& info) const in CreateInstanceNormalization()
893 return std::make_unique<RefInstanceNormalizationWorkload>(descriptor, info); in CreateInstanceNormalization()
897 const WorkloadInfo& info) const in CreateL2Normalization()
899 return std::make_unique<RefL2NormalizationWorkload>(descriptor, info); in CreateL2Normalization()
903 const WorkloadInfo& info) const in CreateLogicalBinary()
905 return std::make_unique<RefLogicalBinaryWorkload>(descriptor, info); in CreateLogicalBinary()
909 const WorkloadInfo& info) const in CreateLogSoftmax()
911 return std::make_unique<RefLogSoftmaxWorkload>(descriptor, info); in CreateLogSoftmax()
915 const WorkloadInfo& info) const in CreateLstm()
917 return std::make_unique<RefLstmWorkload>(descriptor, info); in CreateLstm()
921 const WorkloadInfo& info) const in CreateMaximum()
923 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateMaximum()
925 return std::make_unique<RefMaximumWorkload<int32_t>>(descriptor, info); in CreateMaximum()
929 return std::make_unique<RefMaximumWorkload<float>>(descriptor, info); in CreateMaximum()
934 const WorkloadInfo& info) const in CreateMean()
936 return std::make_unique<RefMeanWorkload>(descriptor, info); in CreateMean()
940 const WorkloadInfo& info) const in CreateMemCopy()
946 return std::make_unique<CopyMemGenericWorkload>(descriptor, info); in CreateMemCopy()
950 const WorkloadInfo& info) const in CreateMemImport()
956 return std::make_unique<ImportMemGenericWorkload>(descriptor, info); in CreateMemImport()
960 const WorkloadInfo& info) const in CreateMinimum()
962 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateMinimum()
964 return std::make_unique<RefMinimumWorkload<int32_t>>(descriptor, info); in CreateMinimum()
968 return std::make_unique<RefMinimumWorkload<float>>(descriptor, info); in CreateMinimum()
973 const WorkloadInfo& info) const in CreateMultiplication()
975 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateMultiplication()
977 return std::make_unique<RefMultiplicationWorkload<int32_t>>(descriptor, info); in CreateMultiplication()
981 return std::make_unique<RefMultiplicationWorkload<float>>(descriptor, info); in CreateMultiplication()
986 const WorkloadInfo& info) const in CreateNormalization()
988 return std::make_unique<RefNormalizationWorkload>(descriptor, info); in CreateNormalization()
992 const WorkloadInfo& info) const in CreateOutput()
994 if (info.m_InputTensorInfos.empty() ) in CreateOutput()
998 if (info.m_OutputTensorInfos.empty()) in CreateOutput()
1002 if (info.m_InputTensorInfos[0].GetNumBytes() != info.m_OutputTensorInfos[0].GetNumBytes()) in CreateOutput()
1007 return std::make_unique<CopyMemGenericWorkload>(descriptor, info); in CreateOutput()
1011 const WorkloadInfo& info) const in CreatePad()
1013 return std::make_unique<RefPadWorkload>(descriptor, info); in CreatePad()
1017 const WorkloadInfo& info) const in CreatePermute()
1019 if (IsQSymmS16(info)) in CreatePermute()
1021 return std::make_unique<RefPermuteQSymm16Workload>(descriptor, info); in CreatePermute()
1023 else if (IsBFloat16(info)) in CreatePermute()
1025 return std::make_unique<RefPermuteBFloat16Workload>(descriptor, info); in CreatePermute()
1027 else if (IsQAsymmS8(info)) in CreatePermute()
1029 return std::make_unique<RefPermuteQAsymmS8Workload>(descriptor, info); in CreatePermute()
1032 NullWorkload, NullWorkload, NullWorkload>(descriptor, info); in CreatePermute()
1036 const WorkloadInfo& info) const in CreatePooling2d()
1038 return std::make_unique<RefPooling2dWorkload>(descriptor, info); in CreatePooling2d()
1042 const WorkloadInfo& info) const in CreatePooling3d()
1044 return std::make_unique<RefPooling3dWorkload>(descriptor, info); in CreatePooling3d()
1048 const WorkloadInfo& /*info*/) const in CreatePreCompiled()
1054 const WorkloadInfo& info) const in CreatePrelu()
1056 return std::make_unique<RefPreluWorkload>(descriptor, info); in CreatePrelu()
1060 const WorkloadInfo& info) const in CreateQLstm()
1062 return std::make_unique<RefQLstmWorkload>(descriptor, info); in CreateQLstm()
1066 const WorkloadInfo& info) const in CreateQuantize()
1068 return std::make_unique<RefQuantizeWorkload>(descriptor, info); in CreateQuantize()
1072 const WorkloadInfo& info) const in CreateRank()
1074 return std::make_unique<RefRankWorkload>(descriptor, info); in CreateRank()
1078 const WorkloadInfo& info) const in CreateReduce()
1080 return std::make_unique<RefReduceWorkload>(descriptor, info); in CreateReduce()
1084 const WorkloadInfo& info) const in CreateReshape()
1086 return std::make_unique<RefReshapeWorkload>(descriptor, info); in CreateReshape()
1090 const WorkloadInfo& info) const in CreateResize()
1092 return std::make_unique<RefResizeWorkload>(descriptor, info); in CreateResize()
1096 const WorkloadInfo& info) const in CreateShape()
1098 return std::make_unique<RefShapeWorkload>(descriptor, info); in CreateShape()
1102 const WorkloadInfo& info) const in CreateSlice()
1104 return std::make_unique<RefSliceWorkload>(descriptor, info); in CreateSlice()
1108 const WorkloadInfo& info) const in CreateSoftmax()
1110 return std::make_unique<RefSoftmaxWorkload>(descriptor, info); in CreateSoftmax()
1114 const WorkloadInfo& info) const in CreateSpaceToBatchNd()
1116 return std::make_unique<RefSpaceToBatchNdWorkload>(descriptor, info); in CreateSpaceToBatchNd()
1120 const WorkloadInfo& info) const in CreateSpaceToDepth()
1122 return std::make_unique<RefSpaceToDepthWorkload>(descriptor, info); in CreateSpaceToDepth()
1126 const WorkloadInfo& info) const in CreateSplitter()
1128 return std::make_unique<RefSplitterWorkload>(descriptor, info); in CreateSplitter()
1132 const WorkloadInfo& info) const in CreateStack()
1134 return std::make_unique<RefStackWorkload>(descriptor, info); in CreateStack()
1138 const WorkloadInfo& info) const in CreateStridedSlice()
1140 return std::make_unique<RefStridedSliceWorkload>(descriptor, info); in CreateStridedSlice()
1144 const WorkloadInfo& info) const in CreateSubtraction()
1146 if (info.m_InputTensorInfos[0].GetDataType() == armnn::DataType::Signed32) in CreateSubtraction()
1148 return std::make_unique<RefSubtractionWorkload<int32_t>>(descriptor, info); in CreateSubtraction()
1152 return std::make_unique<RefSubtractionWorkload<float>>(descriptor, info); in CreateSubtraction()
1157 const WorkloadInfo& info) const in CreateTranspose()
1159 if (IsQSymmS16(info)) in CreateTranspose()
1161 return std::make_unique<RefTransposeQSymm16Workload>(descriptor, info); in CreateTranspose()
1163 else if (IsBFloat16(info)) in CreateTranspose()
1165 return std::make_unique<RefTransposeBFloat16Workload>(descriptor, info); in CreateTranspose()
1167 else if (IsQAsymmS8(info)) in CreateTranspose()
1169 return std::make_unique<RefTransposeQAsymmS8Workload>(descriptor, info); in CreateTranspose()
1172 NullWorkload, NullWorkload, NullWorkload>(descriptor, info); in CreateTranspose()
1177 const WorkloadInfo& info) const in CreateTransposeConvolution2d()
1179 return std::make_unique<RefTransposeConvolution2dWorkload>(descriptor, info); in CreateTransposeConvolution2d()
1184 const WorkloadInfo& info) const in CreateUnidirectionalSequenceLstm()
1186 return std::make_unique<RefUnidirectionalSequenceLstmWorkload>(descriptor, info);; in CreateUnidirectionalSequenceLstm()