diff options
author | Derek Lamberti <derek.lamberti@arm.com> | 2019-11-13 10:36:08 +0000 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-11-14 12:52:23 +0000 |
commit | e3803eae5f8866e536b399ef0003df8d2b6cb03d (patch) | |
tree | 96af5b88a380aa8e990aee020d1e78b964799bd8 | |
parent | 199c98eaa76d35d1122a02596b3a1db93b07c22c (diff) | |
download | armnn-e3803eae5f8866e536b399ef0003df8d2b6cb03d.tar.gz |
CL & Neon workload factories inherit from WorkloadFactoryBase
Change-Id: I1f694be7ef1d333b5ef9b60ea7029454ade02628
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
-rw-r--r-- | src/backends/backendsCommon/CMakeLists.txt | 1 | ||||
-rw-r--r-- | src/backends/backendsCommon/WorkloadFactoryBase.hpp | 256 | ||||
-rw-r--r-- | src/backends/cl/ClWorkloadFactory.hpp | 3 | ||||
-rw-r--r-- | src/backends/neon/NeonWorkloadFactory.hpp | 3 |
4 files changed, 261 insertions, 2 deletions
diff --git a/src/backends/backendsCommon/CMakeLists.txt b/src/backends/backendsCommon/CMakeLists.txt index d4947943ae..8c1cdce587 100644 --- a/src/backends/backendsCommon/CMakeLists.txt +++ b/src/backends/backendsCommon/CMakeLists.txt @@ -39,6 +39,7 @@ list(APPEND armnnBackendsCommon_sources WorkloadData.hpp WorkloadFactory.cpp WorkloadFactory.hpp + WorkloadFactoryBase.hpp Workload.hpp WorkloadInfo.hpp WorkloadUtils.cpp diff --git a/src/backends/backendsCommon/WorkloadFactoryBase.hpp b/src/backends/backendsCommon/WorkloadFactoryBase.hpp new file mode 100644 index 0000000000..6f7437413a --- /dev/null +++ b/src/backends/backendsCommon/WorkloadFactoryBase.hpp @@ -0,0 +1,256 @@ +// +// Copyright © 2019 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#pragma once + +#include "WorkloadFactory.hpp" + +namespace armnn +{ + +class WorkloadFactoryBase : public IWorkloadFactory +{ +public: + bool SupportsSubTensors() const override + { return false; }; + + std::unique_ptr<ITensorHandle> CreateSubTensorHandle(ITensorHandle& parent, TensorShape const& subTensorShape, + unsigned int const *subTensorOrigin) const override + { return nullptr; }; + + std::unique_ptr<IWorkload> CreateInput(const InputQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<ITensorHandle> CreateTensorHandle(const TensorInfo& tensorInfo, + const bool IsMemoryManaged = true) const override + { return nullptr; } + + std::unique_ptr<ITensorHandle> CreateTensorHandle(const TensorInfo& tensorInfo, DataLayout dataLayout, + const bool IsMemoryManaged = true) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateAbs(const AbsQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateActivation(const ActivationQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateAddition(const AdditionQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateArgMinMax(const ArgMinMaxQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateBatchNormalization(const BatchNormalizationQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateBatchToSpaceNd(const BatchToSpaceNdQueueDescriptor& descriptor, + const WorkloadInfo& Info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateComparison(const ComparisonQueueDescriptor& descriptor, + const WorkloadInfo& Info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateConcat(const ConcatQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateConstant(const ConstantQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateConvertFp16ToFp32(const ConvertFp16ToFp32QueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateConvertFp32ToFp16(const ConvertFp32ToFp16QueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateConvolution2d(const Convolution2dQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateDebug(const DebugQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateDepthToSpace(const DepthToSpaceQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateDepthwiseConvolution2d(const DepthwiseConvolution2dQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateDequantize(const DequantizeQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateDetectionPostProcess(const DetectionPostProcessQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateDivision(const DivisionQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateFakeQuantization(const FakeQuantizationQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateFloor(const FloorQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateFullyConnected(const FullyConnectedQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateGather(const GatherQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateInstanceNormalization(const InstanceNormalizationQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateL2Normalization(const L2NormalizationQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateLogSoftmax(const LogSoftmaxQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateLstm(const LstmQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateMaximum(const MaximumQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateMean(const MeanQueueDescriptor& descriptor, + const WorkloadInfo& Info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateMemCopy(const MemCopyQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateMemImport(const MemImportQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateMerge(const MergeQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateMinimum(const MinimumQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateMultiplication(const MultiplicationQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateNormalization(const NormalizationQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateOutput(const OutputQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreatePad(const PadQueueDescriptor& descriptor, + const WorkloadInfo& Info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreatePermute(const PermuteQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreatePooling2d(const Pooling2dQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreatePreCompiled(const PreCompiledQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreatePrelu(const PreluQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateQuantize(const QuantizeQueueDescriptor& descriptor, + const WorkloadInfo& Info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateQuantizedLstm(const QuantizedLstmQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateReshape(const ReshapeQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateResize(const ResizeQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateRsqrt(const RsqrtQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateSlice(const SliceQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateSoftmax(const SoftmaxQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateSpaceToBatchNd(const SpaceToBatchNdQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateSpaceToDepth(const SpaceToDepthQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateSubtraction(const SubtractionQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateSplitter(const SplitterQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateStack(const StackQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateStridedSlice(const StridedSliceQueueDescriptor& descriptor, + const WorkloadInfo& Info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateSwitch(const SwitchQueueDescriptor& descriptor, + const WorkloadInfo& Info) const override + { return nullptr; } + + std::unique_ptr<IWorkload> CreateTransposeConvolution2d(const TransposeConvolution2dQueueDescriptor& descriptor, + const WorkloadInfo& info) const override + { return nullptr; } +}; + +} //namespace armnn
\ No newline at end of file diff --git a/src/backends/cl/ClWorkloadFactory.hpp b/src/backends/cl/ClWorkloadFactory.hpp index 18d12bedf2..6c3935d840 100644 --- a/src/backends/cl/ClWorkloadFactory.hpp +++ b/src/backends/cl/ClWorkloadFactory.hpp @@ -7,6 +7,7 @@ #include <armnn/IRuntime.hpp> #include <armnn/Optional.hpp> +#include <backendsCommon/WorkloadFactoryBase.hpp> #include <backendsCommon/OutputHandler.hpp> #include <aclCommon/BaseMemoryManager.hpp> @@ -14,7 +15,7 @@ namespace armnn { // ARM Compute OpenCL workload factory. -class ClWorkloadFactory : public IWorkloadFactory +class ClWorkloadFactory : public WorkloadFactoryBase { public: ClWorkloadFactory(const std::shared_ptr<ClMemoryManager>& memoryManager); diff --git a/src/backends/neon/NeonWorkloadFactory.hpp b/src/backends/neon/NeonWorkloadFactory.hpp index 5bee771528..f876092c45 100644 --- a/src/backends/neon/NeonWorkloadFactory.hpp +++ b/src/backends/neon/NeonWorkloadFactory.hpp @@ -6,6 +6,7 @@ #include <armnn/Optional.hpp> +#include <backendsCommon/WorkloadFactoryBase.hpp> #include <backendsCommon/OutputHandler.hpp> #include <aclCommon/BaseMemoryManager.hpp> @@ -15,7 +16,7 @@ namespace armnn { // Neon workload factory. -class NeonWorkloadFactory : public IWorkloadFactory +class NeonWorkloadFactory : public WorkloadFactoryBase { public: NeonWorkloadFactory(const std::shared_ptr<NeonMemoryManager>& memoryManager); |