29 #include <arm_compute/core/CL/CLKernelLibrary.h> 30 #include <arm_compute/runtime/CL/CLBufferAllocator.h> 31 #include <arm_compute/runtime/CL/CLScheduler.h> 48 std::string& outReasonIfUnsupported)
55 std::string& outReasonIfUnsupported,
71 if (modelOptions->SaveCachedNetwork())
75 auto cachedFd = modelOptions->GetCachedFileDescriptor();
78 std::vector<uint8_t> compiledContextData;
79 std::stringstream stream;
83 std::string
const serializedString{stream.str()};
84 std::copy(serializedString.begin(),
85 serializedString.end(),
86 std::back_inserter(compiledContextData));
87 auto success = write(cachedFd, compiledContextData.data(), compiledContextData.size());
90 ARMNN_LOG(
info) <<
"ClWorkloadFactory:: Could not cache the compiled context!";
96 auto filePath = modelOptions->GetCachedNetworkFilePath();
97 if (filePath !=
"" && fs::exists(filePath) && fs::is_regular_file(filePath))
100 std::ofstream file(filePath, std::ios::out | std::ios::binary);
108 std::unique_ptr<IWorkload> ClWorkloadFactory::MakeWorkload(
const QueueDescriptorType& descriptor,
114 return MakeWorkloadHelper<FloatWorkload, Uint8Workload>(descriptor,
info, std::forward<Args>(args)...);
122 template <
typename Workload,
typename QueueDescriptorType,
typename... Args>
123 std::unique_ptr<IWorkload> ClWorkloadFactory::MakeWorkload(
const QueueDescriptorType& descriptor,
129 return std::make_unique<Workload>(descriptor,
info, std::forward<Args>(args)...);
137 void ClWorkloadFactory::InitializeCLCompileContext()
140 auto context = arm_compute::CLKernelLibrary::get().context();
141 auto device = arm_compute::CLKernelLibrary::get().get_device();
142 m_CLCompileContext = arm_compute::CLCompileContext(context, device);
144 if (m_ModelContextPtr)
149 if (!(modelOptions->SaveCachedNetwork()))
152 auto cachedFd = modelOptions->GetCachedFileDescriptor();
155 struct stat statBuffer;
156 if (fstat(cachedFd, &statBuffer) == 0)
158 long dataSize =
static_cast<long>(statBuffer.st_size);
161 auto offset = lseek(cachedFd, 0, SEEK_CUR);
164 std::vector <uint8_t> compiledContextData(static_cast<unsigned int>(dataSize));
165 auto success = pread(cachedFd, compiledContextData.data(), compiledContextData.size(), 0);
171 compiledContextData);
179 if (filePath !=
"" && fs::exists(filePath) && fs::is_regular_file(filePath))
182 deserializer.
Deserialize(m_CLCompileContext, context, device, filePath);
189 : m_MemoryManager(memoryManager), m_ModelContextPtr(
IBackendInternal::IBackendSpecificModelContextPtr{})
191 InitializeCLCompileContext();
196 : m_MemoryManager(memoryManager), m_ModelContextPtr(modelContextPtr)
198 InitializeCLCompileContext();
202 const bool IsMemoryManaged)
const 205 std::unique_ptr<ClTensorHandle> tensorHandle = std::make_unique<ClTensorHandle>(tensorInfo);
206 tensorHandle->SetMemoryGroup(m_MemoryManager->GetInterLayerMemoryGroup());
213 const bool IsMemoryManaged)
const 216 std::unique_ptr<ClTensorHandle> tensorHandle = std::make_unique<ClTensorHandle>(tensorInfo, dataLayout);
217 tensorHandle->SetMemoryGroup(m_MemoryManager->GetInterLayerMemoryGroup());
224 unsigned int const* subTensorOrigin)
const 227 arm_compute::TensorShape shape = armcomputetensorutils::BuildArmComputeTensorShape(subTensorShape);
234 coords.set(i, armnn::numeric_cast<int>(subTensorOrigin[revertedIndex]));
237 const arm_compute::TensorShape parentShape = armcomputetensorutils::BuildArmComputeTensorShape(parent.
GetShape());
238 if (!::arm_compute::error_on_invalid_subtensor(__func__, __FILE__, __LINE__, parentShape, coords, shape))
243 return std::make_unique<ClSubTensorHandle>(
244 PolymorphicDowncast<IClTensorHandle*>(&parent), shape, coords);
255 auto activationQueueDescriptor = PolymorphicDowncast<const ActivationQueueDescriptor*>(&descriptor);
256 return MakeWorkload<ClActivationWorkload>(*activationQueueDescriptor,
info, m_CLCompileContext);
260 auto additionQueueDescriptor = PolymorphicDowncast<const AdditionQueueDescriptor*>(&descriptor);
261 return MakeWorkload<ClAdditionWorkload>(*additionQueueDescriptor,
info, m_CLCompileContext);
265 auto argMinMaxQueueDescriptor = PolymorphicDowncast<const ArgMinMaxQueueDescriptor*>(&descriptor);
266 return std::make_unique<ClArgMinMaxWorkload>(*argMinMaxQueueDescriptor,
info, m_CLCompileContext);
270 auto batchNormalizationQueueDescriptor
271 = PolymorphicDowncast<const BatchNormalizationQueueDescriptor*>(&descriptor);
272 return MakeWorkload<ClBatchNormalizationFloatWorkload, NullWorkload>
273 (*batchNormalizationQueueDescriptor,
info, m_CLCompileContext);
277 auto batchToSpaceNdQueueDescriptor
278 = PolymorphicDowncast<const BatchToSpaceNdQueueDescriptor*>(&descriptor);
279 return MakeWorkload<ClBatchToSpaceNdWorkload>(*batchToSpaceNdQueueDescriptor,
info, m_CLCompileContext);
283 auto castQueueDescriptor = PolymorphicDowncast<const CastQueueDescriptor*>(&descriptor);
284 return MakeWorkload<ClCastWorkload>(*castQueueDescriptor,
info, m_CLCompileContext);
288 auto channelShuffleQueueDescriptor
289 = PolymorphicDowncast<const ChannelShuffleQueueDescriptor*>(&descriptor);
290 return MakeWorkload<ClChannelShuffleWorkload>(*channelShuffleQueueDescriptor,
info, m_CLCompileContext);
294 auto comparisonQueueDescriptor = PolymorphicDowncast<const ComparisonQueueDescriptor*>(&descriptor);
295 return MakeWorkload<ClComparisonWorkload>(*comparisonQueueDescriptor,
info, m_CLCompileContext);
299 auto concatQueueDescriptor = PolymorphicDowncast<const ConcatQueueDescriptor*>(&descriptor);
300 return MakeWorkload<ClConcatWorkload>(*concatQueueDescriptor,
info, m_CLCompileContext);
304 auto constantQueueDescriptor = PolymorphicDowncast<const ConstantQueueDescriptor*>(&descriptor);
305 return MakeWorkload<ClConstantWorkload>(*constantQueueDescriptor,
info, m_CLCompileContext);
309 auto convertFp16ToFp32QueueDescriptor
310 = PolymorphicDowncast<const ConvertFp16ToFp32QueueDescriptor*>(&descriptor);
311 return MakeWorkload<ClConvertFp16ToFp32Workload>(*convertFp16ToFp32QueueDescriptor,
317 auto convertFp32ToFp16QueueDescriptor
318 = PolymorphicDowncast<const ConvertFp32ToFp16QueueDescriptor*>(&descriptor);
319 return MakeWorkload<ClConvertFp32ToFp16Workload>(*convertFp32ToFp16QueueDescriptor,
325 auto convolution2dQueueDescriptor = PolymorphicDowncast<const Convolution2dQueueDescriptor*>(&descriptor);
327 bool isFastMathEnabled =
false;
328 if (m_ModelContextPtr)
330 if (m_ModelContextPtr.get() !=
nullptr)
339 return MakeWorkload<ClConvolution2dWorkload>(*convolution2dQueueDescriptor,
341 m_MemoryManager->GetIntraLayerManager(),
347 auto convolution3dQueueDescriptor = PolymorphicDowncast<const Convolution3dQueueDescriptor*>(&descriptor);
349 bool isFastMathEnabled =
false;
350 if (m_ModelContextPtr)
352 if (m_ModelContextPtr.get() !=
nullptr)
361 return MakeWorkload<ClConvolution3dWorkload>(*convolution3dQueueDescriptor,
363 m_MemoryManager->GetIntraLayerManager(),
369 auto debugQueueDescriptor = PolymorphicDowncast<const DebugQueueDescriptor*>(&descriptor);
370 return MakeWorkload<NullWorkload, NullWorkload>(*debugQueueDescriptor,
info, m_CLCompileContext);
374 auto depthToSpaceQueueDescriptor = PolymorphicDowncast<const DepthToSpaceQueueDescriptor*>(&descriptor);
375 return MakeWorkload<ClDepthToSpaceWorkload>(*depthToSpaceQueueDescriptor,
info, m_CLCompileContext);
379 auto depthwiseConvolution2dQueueDescriptor
380 = PolymorphicDowncast<const DepthwiseConvolution2dQueueDescriptor*>(&descriptor);
381 return MakeWorkload<ClDepthwiseConvolutionWorkload>(*depthwiseConvolution2dQueueDescriptor,
387 auto dequantizeQueueDescriptor = PolymorphicDowncast<const DequantizeQueueDescriptor*>(&descriptor);
388 return MakeWorkload<ClDequantizeWorkload>(*dequantizeQueueDescriptor,
info, m_CLCompileContext);
392 auto detectionPostProcessQueueDescriptor
393 = PolymorphicDowncast<const DetectionPostProcessQueueDescriptor*>(&descriptor);
394 return MakeWorkload<NullWorkload, NullWorkload>(*detectionPostProcessQueueDescriptor,
400 auto divisionQueueDescriptor = PolymorphicDowncast<const DivisionQueueDescriptor*>(&descriptor);
401 return std::make_unique<ClDivisionWorkload>(*divisionQueueDescriptor,
info, m_CLCompileContext);
405 auto elementwiseUnaryQueueDescriptor
406 = PolymorphicDowncast<const ElementwiseUnaryQueueDescriptor*>(&descriptor);
408 switch(elementwiseUnaryQueueDescriptor->m_Parameters.m_Operation)
413 absQueueDescriptor.
m_Inputs = elementwiseUnaryQueueDescriptor->m_Inputs;
414 absQueueDescriptor.
m_Outputs = elementwiseUnaryQueueDescriptor->m_Outputs;
416 return std::make_unique<ClAbsWorkload>(absQueueDescriptor,
info, m_CLCompileContext);
419 return std::make_unique<ClExpWorkload>(*elementwiseUnaryQueueDescriptor,
info, m_CLCompileContext);
421 return std::make_unique<ClLogWorkload>(*elementwiseUnaryQueueDescriptor,
info, m_CLCompileContext);
423 return std::make_unique<ClLogicalNotWorkload>(*elementwiseUnaryQueueDescriptor,
427 return std::make_unique<ClNegWorkload>(*elementwiseUnaryQueueDescriptor,
info, m_CLCompileContext);
431 rsqrtQueueDescriptor.
m_Inputs = elementwiseUnaryQueueDescriptor->m_Inputs;
432 rsqrtQueueDescriptor.
m_Outputs = elementwiseUnaryQueueDescriptor->m_Outputs;
434 return std::make_unique<ClRsqrtWorkload>(rsqrtQueueDescriptor,
info, m_CLCompileContext);
437 return std::make_unique<ClSinWorkload>(*elementwiseUnaryQueueDescriptor,
info, m_CLCompileContext);
444 auto fillQueueDescriptor = PolymorphicDowncast<const FillQueueDescriptor*>(&descriptor);
445 return std::make_unique<ClFillWorkload>(*fillQueueDescriptor,
info, m_CLCompileContext);
449 auto floorQueueDescriptor = PolymorphicDowncast<const FloorQueueDescriptor*>(&descriptor);
450 return MakeWorkload<ClFloorFloatWorkload, NullWorkload>(*floorQueueDescriptor,
info, m_CLCompileContext);
454 auto fullyConnectedQueueDescriptor
455 = PolymorphicDowncast<const FullyConnectedQueueDescriptor*>(&descriptor);
456 return MakeWorkload<ClFullyConnectedWorkload>(*fullyConnectedQueueDescriptor,
458 m_MemoryManager->GetIntraLayerManager(),
463 auto gatherQueueDescriptor = PolymorphicDowncast<const GatherQueueDescriptor*>(&descriptor);
464 return MakeWorkload<ClGatherWorkload>(*gatherQueueDescriptor,
info, m_CLCompileContext);
468 auto inputQueueDescriptor = PolymorphicDowncast<const InputQueueDescriptor*>(&descriptor);
469 return std::make_unique<CopyMemGenericWorkload>(*inputQueueDescriptor,
info);
473 auto instanceNormalizationQueueDescriptor
474 = PolymorphicDowncast<const InstanceNormalizationQueueDescriptor*>(&descriptor);
475 return MakeWorkload<ClInstanceNormalizationWorkload>(*instanceNormalizationQueueDescriptor,
481 auto l2NormalizationQueueDescriptor
482 = PolymorphicDowncast<const L2NormalizationQueueDescriptor*>(&descriptor);
483 return MakeWorkload<ClL2NormalizationFloatWorkload, NullWorkload>(*l2NormalizationQueueDescriptor,
489 auto logicalBinaryQueueDescriptor = PolymorphicDowncast<const LogicalBinaryQueueDescriptor*>(&descriptor);
491 switch(logicalBinaryQueueDescriptor->m_Parameters.m_Operation)
494 return std::make_unique<ClLogicalAndWorkload>(*logicalBinaryQueueDescriptor,
498 return std::make_unique<ClLogicalOrWorkload>(*logicalBinaryQueueDescriptor,
507 auto logSoftmaxQueueDescriptor = PolymorphicDowncast<const LogSoftmaxQueueDescriptor*>(&descriptor);
509 return MakeWorkload<ClLogSoftmaxWorkload>(*logSoftmaxQueueDescriptor,
511 m_MemoryManager->GetIntraLayerManager(),
516 auto lstmQueueDescriptor = PolymorphicDowncast<const LstmQueueDescriptor*>(&descriptor);
517 return MakeWorkload<ClLstmFloatWorkload, NullWorkload>(*lstmQueueDescriptor,
info, m_CLCompileContext);
521 auto maximumQueueDescriptor = PolymorphicDowncast<const MaximumQueueDescriptor*>(&descriptor);
522 return MakeWorkload<ClMaximumWorkload>(*maximumQueueDescriptor,
info, m_CLCompileContext);
526 auto meanQueueDescriptor = PolymorphicDowncast<const MeanQueueDescriptor*>(&descriptor);
527 return MakeWorkload<ClMeanWorkload>(*meanQueueDescriptor,
info, m_CLCompileContext);
531 auto memCopyQueueDescriptor = PolymorphicDowncast<const MemCopyQueueDescriptor*>(&descriptor);
532 if (memCopyQueueDescriptor->m_Inputs.empty() || !memCopyQueueDescriptor->m_Inputs[0])
536 return MakeWorkload<CopyMemGenericWorkload>(*memCopyQueueDescriptor,
info);
540 auto memImportQueueDescriptor = PolymorphicDowncast<const MemImportQueueDescriptor*>(&descriptor);
541 if (memImportQueueDescriptor->m_Inputs.empty() || !memImportQueueDescriptor->m_Inputs[0])
545 return std::make_unique<ImportMemGenericWorkload>(*memImportQueueDescriptor,
info);
549 auto minimumQueueDescriptor = PolymorphicDowncast<const MinimumQueueDescriptor*>(&descriptor);
550 return MakeWorkload<ClMinimumWorkload>(*minimumQueueDescriptor,
info, m_CLCompileContext);
554 auto multiplicationQueueDescriptor = PolymorphicDowncast<const MultiplicationQueueDescriptor*>(&descriptor);
555 return MakeWorkload<ClMultiplicationWorkload>(*multiplicationQueueDescriptor,
info, m_CLCompileContext);
559 auto normalizationQueueDescriptor = PolymorphicDowncast<const NormalizationQueueDescriptor*>(&descriptor);
560 return MakeWorkload<ClNormalizationFloatWorkload, NullWorkload>(*normalizationQueueDescriptor,
566 auto outputQueueDescriptor = PolymorphicDowncast<const OutputQueueDescriptor*>(&descriptor);
567 return std::make_unique<CopyMemGenericWorkload>(*outputQueueDescriptor,
info);
571 auto padQueueDescriptor = PolymorphicDowncast<const PadQueueDescriptor*>(&descriptor);
572 return MakeWorkload<ClPadWorkload>(*padQueueDescriptor,
info, m_CLCompileContext);
576 auto permuteQueueDescriptor = PolymorphicDowncast<const PermuteQueueDescriptor*>(&descriptor);
577 return MakeWorkload<ClPermuteWorkload>(*permuteQueueDescriptor,
info, m_CLCompileContext);
581 auto pooling2dQueueDescriptor = PolymorphicDowncast<const Pooling2dQueueDescriptor*>(&descriptor);
582 return MakeWorkload<ClPooling2dWorkload>(*pooling2dQueueDescriptor,
info, m_CLCompileContext);
586 auto preCompiledQueueDescriptor = PolymorphicDowncast<const PreCompiledQueueDescriptor*>(&descriptor);
587 return MakeWorkload<NullWorkload, NullWorkload>(*preCompiledQueueDescriptor,
info, m_CLCompileContext);
591 auto preluQueueDescriptor = PolymorphicDowncast<const PreluQueueDescriptor*>(&descriptor);
592 return MakeWorkload<ClPreluWorkload>(*preluQueueDescriptor,
info, m_CLCompileContext);
596 auto qLstmQueueDescriptor = PolymorphicDowncast<const QLstmQueueDescriptor*>(&descriptor);
597 return std::make_unique<ClQLstmWorkload>(*qLstmQueueDescriptor,
info, m_CLCompileContext);
601 auto quantizeQueueDescriptor = PolymorphicDowncast<const QuantizeQueueDescriptor*>(&descriptor);
602 return MakeWorkload<ClQuantizeWorkload>(*quantizeQueueDescriptor,
info, m_CLCompileContext);
606 auto quantizedLstmQueueDescriptor = PolymorphicDowncast<const QuantizedLstmQueueDescriptor*>(&descriptor);
607 return MakeWorkload<ClQuantizedLstmWorkload>(*quantizedLstmQueueDescriptor,
info, m_CLCompileContext);
611 auto rankQueueDescriptor = PolymorphicDowncast<const RankQueueDescriptor*>(&descriptor);
612 return std::make_unique<ClRankWorkload>(*rankQueueDescriptor,
info);
616 auto reduceQueueDescriptor = PolymorphicDowncast<const ReduceQueueDescriptor*>(&descriptor);
617 return std::make_unique<ClReduceWorkload>(*reduceQueueDescriptor,
info);
621 auto reshapeQueueDescriptor = PolymorphicDowncast<const ReshapeQueueDescriptor*>(&descriptor);
622 return MakeWorkload<ClReshapeWorkload>(*reshapeQueueDescriptor,
info, m_CLCompileContext);
626 auto resizeQueueDescriptor = PolymorphicDowncast<const ResizeQueueDescriptor*>(&descriptor);
627 return MakeWorkload<ClResizeWorkload>(*resizeQueueDescriptor,
info, m_CLCompileContext);
631 auto sliceQueueDescriptor = PolymorphicDowncast<const SliceQueueDescriptor*>(&descriptor);
632 return MakeWorkload<ClSliceWorkload>(*sliceQueueDescriptor,
info, m_CLCompileContext);
636 auto softmaxQueueDescriptor = PolymorphicDowncast<const SoftmaxQueueDescriptor*>(&descriptor);
637 return std::make_unique<ClSoftmaxWorkload>(*softmaxQueueDescriptor,
639 m_MemoryManager->GetIntraLayerManager(),
644 auto spaceToBatchNdQueueDescriptor
645 = PolymorphicDowncast<const SpaceToBatchNdQueueDescriptor*>(&descriptor);
646 return MakeWorkload<ClSpaceToBatchNdWorkload>(*spaceToBatchNdQueueDescriptor,
info, m_CLCompileContext);
650 auto spaceToDepthQueueDescriptor = PolymorphicDowncast<const SpaceToDepthQueueDescriptor*>(&descriptor);
651 return MakeWorkload<ClSpaceToDepthWorkload>(*spaceToDepthQueueDescriptor,
info, m_CLCompileContext);
655 auto splitterQueueDescriptor = PolymorphicDowncast<const SplitterQueueDescriptor*>(&descriptor);
656 return MakeWorkload<ClSplitterWorkload>(*splitterQueueDescriptor,
info, m_CLCompileContext);
660 auto stackQueueDescriptor = PolymorphicDowncast<const StackQueueDescriptor*>(&descriptor);
661 return MakeWorkload<ClStackWorkload>(*stackQueueDescriptor,
info, m_CLCompileContext);
665 auto stridedSliceQueueDescriptor = PolymorphicDowncast<const StridedSliceQueueDescriptor*>(&descriptor);
666 return MakeWorkload<ClStridedSliceWorkload>(*stridedSliceQueueDescriptor,
info, m_CLCompileContext);
670 auto subtractionQueueDescriptor = PolymorphicDowncast<const SubtractionQueueDescriptor*>(&descriptor);
671 return MakeWorkload<ClSubtractionWorkload>(*subtractionQueueDescriptor,
info, m_CLCompileContext);
675 auto transposeQueueDescriptor = PolymorphicDowncast<const TransposeQueueDescriptor*>(&descriptor);
676 return MakeWorkload<ClTransposeWorkload>(*transposeQueueDescriptor,
info, m_CLCompileContext);
680 auto transposeConvolution2dQueueDescriptor
681 = PolymorphicDowncast<const TransposeConvolution2dQueueDescriptor*>(&descriptor);
682 return MakeWorkload<ClTransposeConvolution2dWorkload>(*transposeConvolution2dQueueDescriptor,
684 m_MemoryManager->GetIntraLayerManager(),
695 return MakeWorkload<ClActivationWorkload>(descriptor,
info, m_CLCompileContext);
701 return MakeWorkload<ClAdditionWorkload>(descriptor,
info, m_CLCompileContext);
707 return std::make_unique<ClArgMinMaxWorkload>(descriptor,
info, m_CLCompileContext);
714 return MakeWorkload<ClBatchNormalizationFloatWorkload, NullWorkload>(descriptor,
info, m_CLCompileContext);
720 return MakeWorkload<ClBatchToSpaceNdWorkload>(descriptor,
info, m_CLCompileContext);
723 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateCast(
const CastQueueDescriptor& descriptor,
726 return MakeWorkload<ClCastWorkload>(descriptor,
info, m_CLCompileContext);
732 return MakeWorkload<ClChannelShuffleWorkload>(descriptor,
info, m_CLCompileContext);
738 return MakeWorkload<ClComparisonWorkload>(descriptor,
info, m_CLCompileContext);
744 return MakeWorkload<ClConcatWorkload>(descriptor,
info, m_CLCompileContext);
750 return MakeWorkload<ClConstantWorkload>(descriptor,
info, m_CLCompileContext);
757 return MakeWorkload<ClConvertFp16ToFp32Workload>(descriptor,
info, m_CLCompileContext);
764 return MakeWorkload<ClConvertFp32ToFp16Workload>(descriptor,
info, m_CLCompileContext);
770 bool isFastMathEnabled =
false;
771 if (m_ModelContextPtr)
773 if (m_ModelContextPtr.get() !=
nullptr)
782 return MakeWorkload<ClConvolution2dWorkload>(descriptor,
784 m_MemoryManager->GetIntraLayerManager(),
792 bool isFastMathEnabled =
false;
793 if (m_ModelContextPtr)
795 if (m_ModelContextPtr.get() !=
nullptr)
804 return MakeWorkload<ClConvolution3dWorkload>(descriptor,
806 m_MemoryManager->GetIntraLayerManager(),
814 return MakeWorkload<NullWorkload, NullWorkload>(descriptor,
info, m_CLCompileContext);
820 return MakeWorkload<ClDepthToSpaceWorkload>(descriptor,
info, m_CLCompileContext);
827 return MakeWorkload<ClDepthwiseConvolutionWorkload>(descriptor,
info, m_CLCompileContext);
833 return MakeWorkload<ClDequantizeWorkload>(descriptor,
info, m_CLCompileContext);
840 return MakeWorkload<NullWorkload, NullWorkload>(descriptor,
info, m_CLCompileContext);
846 return std::make_unique<ClDivisionWorkload>(descriptor,
info, m_CLCompileContext);
860 return std::make_unique<ClAbsWorkload>(absQueueDescriptor,
info, m_CLCompileContext);
863 return std::make_unique<ClExpWorkload>(descriptor,
info, m_CLCompileContext);
865 return std::make_unique<ClLogWorkload>(descriptor,
info, m_CLCompileContext);
867 return std::make_unique<ClLogicalNotWorkload>(descriptor,
info, m_CLCompileContext);
869 return std::make_unique<ClNegWorkload>(descriptor,
info, m_CLCompileContext);
873 rsqrtQueueDescriptor.
m_Inputs = descriptor.m_Inputs;
874 rsqrtQueueDescriptor.
m_Outputs = descriptor.m_Outputs;
876 return std::make_unique<ClRsqrtWorkload>(rsqrtQueueDescriptor,
info, m_CLCompileContext);
879 return std::make_unique<ClSinWorkload>(descriptor,
info, m_CLCompileContext);
885 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateFill(
const FillQueueDescriptor& descriptor,
888 return std::make_unique<ClFillWorkload>(descriptor,
info, m_CLCompileContext);
894 return MakeWorkload<ClFloorFloatWorkload, NullWorkload>(descriptor,
info, m_CLCompileContext);
900 return MakeWorkload<ClFullyConnectedWorkload>(descriptor,
902 m_MemoryManager->GetIntraLayerManager(),
909 return MakeWorkload<ClGatherWorkload>(descriptor,
info, m_CLCompileContext);
915 return std::make_unique<CopyMemGenericWorkload>(descriptor,
info);
922 return MakeWorkload<ClInstanceNormalizationWorkload>(descriptor,
info, m_CLCompileContext);
928 return MakeWorkload<ClL2NormalizationFloatWorkload, NullWorkload>(descriptor,
info, m_CLCompileContext);
937 return std::make_unique<ClLogicalAndWorkload>(descriptor,
info, m_CLCompileContext);
939 return std::make_unique<ClLogicalOrWorkload>(descriptor,
info, m_CLCompileContext);
948 return MakeWorkload<ClLogSoftmaxWorkload>(descriptor,
950 m_MemoryManager->GetIntraLayerManager(),
957 return MakeWorkload<ClLstmFloatWorkload, NullWorkload>(descriptor,
info, m_CLCompileContext);
963 return MakeWorkload<ClMaximumWorkload>(descriptor,
info, m_CLCompileContext);
969 return MakeWorkload<ClMeanWorkload>(descriptor,
info, m_CLCompileContext);
980 return MakeWorkload<CopyMemGenericWorkload>(descriptor,
info);
991 return std::make_unique<ImportMemGenericWorkload>(descriptor,
info);
997 return MakeWorkload<ClMinimumWorkload>(descriptor,
info, m_CLCompileContext);
1003 return MakeWorkload<ClMultiplicationWorkload>(descriptor,
info, m_CLCompileContext);
1009 return MakeWorkload<ClNormalizationFloatWorkload, NullWorkload>(descriptor,
info, m_CLCompileContext);
1015 return std::make_unique<CopyMemGenericWorkload>(descriptor,
info);
1021 return MakeWorkload<ClPadWorkload>(descriptor,
info, m_CLCompileContext);
1027 return MakeWorkload<ClPermuteWorkload>(descriptor,
info, m_CLCompileContext);
1033 return MakeWorkload<ClPooling2dWorkload>(descriptor,
info, m_CLCompileContext);
1039 return MakeWorkload<NullWorkload, NullWorkload>(descriptor,
info, m_CLCompileContext);
1045 return MakeWorkload<ClPreluWorkload>(descriptor,
info, m_CLCompileContext);
1048 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateQLstm(
const QLstmQueueDescriptor& descriptor,
1051 return std::make_unique<ClQLstmWorkload>(descriptor,
info, m_CLCompileContext);
1057 return MakeWorkload<ClQuantizeWorkload>(descriptor,
info, m_CLCompileContext);
1063 return MakeWorkload<ClQuantizedLstmWorkload>(descriptor,
info, m_CLCompileContext);
1069 return std::make_unique<ClRankWorkload>(descriptor,
info);
1075 return std::make_unique<ClReduceWorkload>(descriptor,
info);
1081 return MakeWorkload<ClReshapeWorkload>(descriptor,
info, m_CLCompileContext);
1087 return MakeWorkload<ClResizeWorkload>(descriptor,
info, m_CLCompileContext);
1093 return MakeWorkload<ClSliceWorkload>(descriptor,
info, m_CLCompileContext);
1099 return std::make_unique<ClSoftmaxWorkload>(descriptor,
1101 m_MemoryManager->GetIntraLayerManager(),
1102 m_CLCompileContext);
1108 return MakeWorkload<ClSpaceToBatchNdWorkload>(descriptor,
info, m_CLCompileContext);
1114 return MakeWorkload<ClSpaceToDepthWorkload>(descriptor,
info, m_CLCompileContext);
1120 return MakeWorkload<ClSplitterWorkload>(descriptor,
info, m_CLCompileContext);
1126 return MakeWorkload<ClStackWorkload>(descriptor,
info, m_CLCompileContext);
1132 return MakeWorkload<ClStridedSliceWorkload>(descriptor,
info, m_CLCompileContext);
1138 return MakeWorkload<ClSubtractionWorkload>(descriptor,
info, m_CLCompileContext);
1144 return MakeWorkload<ClTransposeWorkload>(descriptor,
info, m_CLCompileContext);
1151 return MakeWorkload<ClTransposeConvolution2dWorkload>(descriptor,
1153 m_MemoryManager->GetIntraLayerManager(),
1154 m_CLCompileContext);
std::unique_ptr< IWorkload > CreateSubtraction(const SubtractionQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< ITensorHandle > CreateSubTensorHandle(ITensorHandle &parent, TensorShape const &subTensorShape, unsigned int const *subTensorOrigin) const override
std::unique_ptr< IWorkload > CreateElementwiseUnary(const ElementwiseUnaryQueueDescriptor &descriptor, const WorkloadInfo &info) const override
ClWorkloadFactory(const std::shared_ptr< ClMemoryManager > &memoryManager)
UnaryOperation m_Operation
Specifies the elementwiseUnary operation to execute.
std::unique_ptr< IWorkload > CreateStridedSlice(const StridedSliceQueueDescriptor &, const WorkloadInfo &) const override
Interface for a layer that is connectable to other layers via InputSlots and OutputSlots.
std::unique_ptr< IWorkload > CreateInput(const InputQueueDescriptor &, const WorkloadInfo &) const override
void AfterWorkloadsCreated() override
std::unique_ptr< IWorkload > CreateL2Normalization(const L2NormalizationQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateBatchNormalization(const BatchNormalizationQueueDescriptor &, const WorkloadInfo &) const override
constexpr const char * ClBackendId()
std::vector< BackendOptions > ModelOptions
std::unique_ptr< IWorkload > CreateSoftmax(const SoftmaxQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateSpaceToBatchNd(const SpaceToBatchNdQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreatePooling2d(const Pooling2dQueueDescriptor &, const WorkloadInfo &) const override
std::array< unsigned int, MaxNumOfTensorDimensions > Coordinates
std::unique_ptr< IWorkload > CreateMean(const MeanQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreatePad(const PadQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< ITensorHandle > CreateTensorHandle(const TensorInfo &tensorInfo, const bool IsMemoryManaged=true) const override
std::unique_ptr< IWorkload > CreateLstm(const LstmQueueDescriptor &, const WorkloadInfo &) const override
const BackendId & GetBackendId() const override
std::unique_ptr< IWorkload > CreateDepthToSpace(const DepthToSpaceQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateLogSoftmax(const LogSoftmaxQueueDescriptor &, const WorkloadInfo &) const override
#define ARMNN_LOG(severity)
std::unique_ptr< IWorkload > CreateComparison(const ComparisonQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateTranspose(const TransposeQueueDescriptor &, const WorkloadInfo &) const override
bool SaveSerializedToStream(std::ostream &stream)
Serializes the ClContext to the stream.
Copyright (c) 2021 ARM Limited and Contributors.
void IgnoreUnused(Ts &&...)
TypedWorkload< QueueDescriptor, armnn::DataType::Float16, armnn::DataType::Float32 > FloatWorkload
std::unique_ptr< IWorkload > CreateDequantize(const DequantizeQueueDescriptor &, const WorkloadInfo &) const override
LayerDescriptor m_Parameters
std::unique_ptr< IWorkload > CreateSplitter(const SplitterQueueDescriptor &, const WorkloadInfo &) const override
LogicalBinaryOperation m_Operation
Specifies the logical operation to execute.
std::unique_ptr< IWorkload > CreateConvertFp32ToFp16(const ConvertFp32ToFp16QueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateNormalization(const NormalizationQueueDescriptor &, const WorkloadInfo &) const override
void DeserializeFromBinary(arm_compute::CLCompileContext &clCompileContext, cl::Context &context, cl::Device &device, const std::vector< uint8_t > &binaryContent)
Deserializes the CLCompileContext built-in programs from binary file contents.
std::unique_ptr< IWorkload > CreateDepthwiseConvolution2d(const DepthwiseConvolution2dQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateInstanceNormalization(const InstanceNormalizationQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateTransposeConvolution2d(const TransposeConvolution2dQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateSpaceToDepth(const SpaceToDepthQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateConcat(const ConcatQueueDescriptor &, const WorkloadInfo &) const override
static bool IsLayerSupported(const Layer &layer, Optional< DataType > dataType, std::string &outReasonIfUnsupported)
std::unique_ptr< IWorkload > CreateWorkload(LayerType type, const QueueDescriptor &descriptor, const WorkloadInfo &info) const override
std::unique_ptr< IWorkload > CreatePrelu(const PreluQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateBatchToSpaceNd(const BatchToSpaceNdQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateArgMinMax(const ArgMinMaxQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreatePermute(const PermuteQueueDescriptor &, const WorkloadInfo &) const override
std::shared_ptr< IBackendModelContext > IBackendSpecificModelContextPtr
std::unique_ptr< IWorkload > CreateDebug(const DebugQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateConstant(const ConstantQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateConvolution2d(const Convolution2dQueueDescriptor &, const WorkloadInfo &) const override
void Serialize(const arm_compute::CLCompileContext &clCompileContext)
Serializes the CLCompileContext built-in programs.
std::unique_ptr< IWorkload > CreateFullyConnected(const FullyConnectedQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateMinimum(const MinimumQueueDescriptor &, const WorkloadInfo &) const override
RuntimeException WrapClError(const cl::Error &clError, const CheckLocation &location)
bool IsFastMathEnabled() const
std::unique_ptr< IWorkload > CreateReshape(const ReshapeQueueDescriptor &, const WorkloadInfo &) const override
static bool IsLayerSupported(const BackendId &backendId, const IConnectableLayer &layer, Optional< DataType > dataType, std::string &outReasonIfUnsupported)
std::unique_ptr< IWorkload > CreateFloor(const FloorQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateRank(const RankQueueDescriptor &, const WorkloadInfo &) const override
virtual TensorShape GetShape() const =0
Get the number of elements for each dimension ordered from slowest iterating dimension to fastest ite...
std::unique_ptr< IWorkload > CreateDetectionPostProcess(const DetectionPostProcessQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateQuantize(const QuantizeQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateAddition(const AdditionQueueDescriptor &, const WorkloadInfo &) const override
void Deserialize(arm_compute::CLCompileContext &clCompileContext, cl::Context &context, cl::Device &device, const std::string &filePath)
Deserializes the CLCompileContext built-in programs from a binary file.
TypedWorkload< QueueDescriptor, armnn::DataType::QAsymmU8 > Uint8Workload
std::unique_ptr< IWorkload > CreateGather(const GatherQueueDescriptor &, const WorkloadInfo &) const override
std::vector< ITensorHandle * > m_Outputs
std::unique_ptr< IWorkload > CreateDivision(const DivisionQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreatePreCompiled(const PreCompiledQueueDescriptor &, const WorkloadInfo &) const override
unsigned int GetNumDimensions() const
Function that returns the tensor rank.
std::unique_ptr< IWorkload > CreateStack(const StackQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateOutput(const OutputQueueDescriptor &, const WorkloadInfo &) const override
Contains information about TensorInfos of a layer.
std::unique_ptr< IWorkload > CreateResize(const ResizeQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateSlice(const SliceQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateConvertFp16ToFp32(const ConvertFp16ToFp32QueueDescriptor &, const WorkloadInfo &) const override
std::vector< ITensorHandle * > m_Inputs
std::unique_ptr< IWorkload > CreateMultiplication(const MultiplicationQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateMaximum(const MaximumQueueDescriptor &, const WorkloadInfo &) const override
std::unique_ptr< IWorkload > CreateActivation(const ActivationQueueDescriptor &, const WorkloadInfo &) const override
The ClBackendModelContext is used to pass in CL specific backend ModelOptions.
std::unique_ptr< IWorkload > CreateMemImport(const MemImportQueueDescriptor &, const WorkloadInfo &) const override
std::string GetCachedNetworkFilePath() const
std::unique_ptr< IWorkload > CreateMemCopy(const MemCopyQueueDescriptor &, const WorkloadInfo &) const override
Depthwise Convolution 2D layer workload data.
std::unique_ptr< IWorkload > CreateQuantizedLstm(const QuantizedLstmQueueDescriptor &, const WorkloadInfo &) const override
LayerType
When adding a new layer, adapt also the LastLayer enum value in the enum class LayerType below...