Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / f9aeef0e036df176699aa96d30d2ca8d7546534e / . / src / backends / neon / NeonWorkloadFactory.hpp
blob: 64951612c148842c06861474d43403d5b56a2e30 [file] [log] [blame]
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]1//
2// Copyright © 2017 Arm Ltd. All rights reserved.
David Beckecb56cd2018-09-05 12:52:57 +0100[diff] [blame]3// SPDX-License-Identifier: MIT
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]4//
5#pragma once
6
David Beck0dbe0ee2018-09-24 15:59:27 +0100[diff] [blame]7#include <backends/OutputHandler.hpp>
Aron Virginas-Tarf9aeef02018-10-12 15:18:03 +0100[diff] [blame^]8#include <backends/aclCommon/memory/BaseMemoryManager.hpp>
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]9
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]10#include <boost/core/ignore_unused.hpp>
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]11#include <boost/optional.hpp>
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]12
13namespace armnn
14{
15
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]16// Neon workload factory.
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]17class NeonWorkloadFactory : public IWorkloadFactory
18{
19public:
surmeh013537c2c2018-05-18 16:31:43 +0100[diff] [blame]20 NeonWorkloadFactory();
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]21
22 virtual Compute GetCompute() const override { return Compute::CpuAcc; }
23
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]24 static bool IsLayerSupported(const Layer& layer, boost::optional<DataType> dataType,
25 std::string& outReasonIfUnsupported);
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]26
27 virtual bool SupportsSubTensors() const override { return true; }
28
29 virtual std::unique_ptr<ITensorHandle> CreateSubTensorHandle(ITensorHandle& parent,
30 TensorShape const& subTensorShape,
31 unsigned int const* subTensorOrigin) const override;
32
33 virtual std::unique_ptr<ITensorHandle> CreateTensorHandle(const TensorInfo& tensorInfo) const override;
34
Francis Murtagh351d13d2018-09-24 15:01:18 +0100[diff] [blame]35 virtual std::unique_ptr<ITensorHandle> CreateTensorHandle(const TensorInfo& tensorInfo,
36 DataLayout dataLayout) const override;
37
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]38 virtual std::unique_ptr<IWorkload> CreateInput(const InputQueueDescriptor& descriptor,
39 const WorkloadInfo& info) const override;
40
41 virtual std::unique_ptr<IWorkload> CreateOutput(const OutputQueueDescriptor& descriptor,
42 const WorkloadInfo& info) const override;
43
44 virtual std::unique_ptr<IWorkload> CreateActivation(const ActivationQueueDescriptor& descriptor,
45 const WorkloadInfo& info) const override;
46
47 virtual std::unique_ptr<IWorkload> CreateSoftmax(const SoftmaxQueueDescriptor& descriptor,
48 const WorkloadInfo& info) const override;
49
50 virtual std::unique_ptr<IWorkload> CreateSplitter(const SplitterQueueDescriptor& descriptor,
51 const WorkloadInfo& info) const override;
52
53 virtual std::unique_ptr<IWorkload> CreateMerger(const MergerQueueDescriptor& descriptor,
54 const WorkloadInfo& info) const override;
55
56 virtual std::unique_ptr<IWorkload> CreateFullyConnected(const FullyConnectedQueueDescriptor& descriptor,
57 const WorkloadInfo& info) const override;
58
59 virtual std::unique_ptr<IWorkload> CreatePermute(const PermuteQueueDescriptor& descriptor,
60 const WorkloadInfo& info) const override;
61
62 virtual std::unique_ptr<IWorkload> CreatePooling2d(const Pooling2dQueueDescriptor& descriptor,
63 const WorkloadInfo& info) const override;
64
65 virtual std::unique_ptr<IWorkload> CreateConvolution2d(const Convolution2dQueueDescriptor& descriptor,
66 const WorkloadInfo& info) const override;
67
68 virtual std::unique_ptr<IWorkload> CreateDepthwiseConvolution2d(
69 const DepthwiseConvolution2dQueueDescriptor& descriptor, const WorkloadInfo& info) const override;
70
71 virtual std::unique_ptr<IWorkload> CreateNormalization(const NormalizationQueueDescriptor& descriptor,
72 const WorkloadInfo& info) const override;
73
74 virtual std::unique_ptr<IWorkload> CreateMultiplication(const MultiplicationQueueDescriptor& descriptor,
75 const WorkloadInfo& info) const override;
76
77 virtual std::unique_ptr<IWorkload> CreateAddition(const AdditionQueueDescriptor& descriptor,
78 const WorkloadInfo& info) const override;
79
80 virtual std::unique_ptr<IWorkload> CreateBatchNormalization(const BatchNormalizationQueueDescriptor& descriptor,
81 const WorkloadInfo& info) const override;
82
83 virtual std::unique_ptr<IWorkload> CreateMemCopy(const MemCopyQueueDescriptor& descriptor,
84 const WorkloadInfo& info) const override;
85
86 virtual std::unique_ptr<IWorkload> CreateResizeBilinear(const ResizeBilinearQueueDescriptor& descriptor,
87 const WorkloadInfo& info) const override;
88
89 virtual std::unique_ptr<IWorkload> CreateFakeQuantization(const FakeQuantizationQueueDescriptor& descriptor,
90 const WorkloadInfo& info) const override;
91
92 virtual std::unique_ptr<IWorkload> CreateL2Normalization(const L2NormalizationQueueDescriptor& descriptor,
93 const WorkloadInfo& info) const override;
94
95 virtual std::unique_ptr<IWorkload> CreateConstant(const ConstantQueueDescriptor& descriptor,
96 const WorkloadInfo& info) const override;
97
98 virtual std::unique_ptr<IWorkload> CreateReshape(const ReshapeQueueDescriptor& descriptor,
99 const WorkloadInfo& info) const override;
100
101 virtual std::unique_ptr<IWorkload> CreateFloor(const FloorQueueDescriptor& descriptor,
102 const WorkloadInfo& info) const override;
surmeh013537c2c2018-05-18 16:31:43 +0100[diff] [blame]103
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]104 virtual std::unique_ptr<IWorkload> CreateLstm(const LstmQueueDescriptor& descriptor,
105 const WorkloadInfo& info) const override;
106
107 virtual std::unique_ptr<IWorkload> CreateConvertFp16ToFp32(const ConvertFp16ToFp32QueueDescriptor& descriptor,
108 const WorkloadInfo& info) const override;
109
110 virtual std::unique_ptr<IWorkload> CreateConvertFp32ToFp16(const ConvertFp32ToFp16QueueDescriptor& descriptor,
111 const WorkloadInfo& info) const override;
112
Francis Murtaghe7a86a42018-08-29 12:42:10 +0100[diff] [blame]113 virtual std::unique_ptr<IWorkload> CreateDivision(const DivisionQueueDescriptor& descriptor,
114 const WorkloadInfo& info) const override;
115
David Beckc2044fe2018-09-05 15:00:38 +0100[diff] [blame]116 virtual std::unique_ptr<IWorkload> CreateSubtraction(const SubtractionQueueDescriptor& descriptor,
117 const WorkloadInfo& info) const override;
118
narpra01a6bf9122018-09-10 09:50:09 +0100[diff] [blame]119 virtual std::unique_ptr<IWorkload> CreateMean(const MeanQueueDescriptor& descriptor,
120 const WorkloadInfo& Info) const override;
121
jimfly012c9322a2018-09-19 10:59:49 +0100[diff] [blame]122 virtual std::unique_ptr<IWorkload> CreatePad(const PadQueueDescriptor& descriptor,
123 const WorkloadInfo& info) const override;
124
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]125 virtual void Finalize() override;
126
127 virtual void Release() override;
128
129 virtual void Acquire() override;
surmeh013537c2c2018-05-18 16:31:43 +0100[diff] [blame]130
131private:
telsoa01c577f2c2018-08-31 09:22:23 +0100[diff] [blame]132#ifdef ARMCOMPUTENEON_ENABLED
133 mutable NeonMemoryManager m_MemoryManager;
134#endif
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]135};
136
137} //namespace armnn