Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / 30cdfcac03fc9f3ab424865b40c0490799e5c8fb / . / src / backends / neon / NeonLayerSupport.hpp
blob: 5724ed85df7e84818c49a57a625c5156622b14ed [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 Beck3cc9a622018-10-12 10:38:31 +0100[diff] [blame]7#include <armnn/ILayerSupport.hpp>
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]8
9namespace armnn
10{
11
arovir014424b0a2018-10-04 10:46:04 +0100[diff] [blame]12class NeonLayerSupport : public ILayerSupport
13{
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]14public:
15 bool IsActivationSupported(const TensorInfo& input,
16 const TensorInfo& output,
17 const ActivationDescriptor& descriptor,
18 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
19
20 bool IsAdditionSupported(const TensorInfo& input0,
21 const TensorInfo& input1,
22 const TensorInfo& output,
23 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
24
25 bool IsBatchNormalizationSupported(const TensorInfo& input,
26 const TensorInfo& output,
27 const TensorInfo& mean,
28 const TensorInfo& var,
29 const TensorInfo& beta,
30 const TensorInfo& gamma,
31 const BatchNormalizationDescriptor& descriptor,
32 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
33
34 bool IsConstantSupported(const TensorInfo& output,
35 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
36
37 bool IsConvertFp16ToFp32Supported(const TensorInfo& input,
38 const TensorInfo& output,
39 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
40
41 bool IsConvertFp32ToFp16Supported(const TensorInfo& input,
42 const TensorInfo& output,
43 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
44
45 bool IsConvolution2dSupported(const TensorInfo& input,
46 const TensorInfo& output,
47 const Convolution2dDescriptor& descriptor,
48 const TensorInfo& weights,
49 const Optional<TensorInfo>& biases,
50 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
51
52 bool IsDepthwiseConvolutionSupported(const TensorInfo& input,
53 const TensorInfo& output,
54 const DepthwiseConvolution2dDescriptor& descriptor,
55 const TensorInfo& weights,
56 const Optional<TensorInfo>& biases,
57 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
58
59 bool IsDivisionSupported(const TensorInfo& input0,
60 const TensorInfo& input1,
61 const TensorInfo& output,
62 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
63
FrancisMurtagh30cdfca2018-12-18 12:57:35 +0000[diff] [blame^]64 bool IsEqualSupported(const TensorInfo& input0,
65 const TensorInfo& input1,
66 const TensorInfo& output,
67 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
68
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]69 bool IsFakeQuantizationSupported(const TensorInfo& input,
70 const FakeQuantizationDescriptor& descriptor,
71 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
72
73 bool IsFloorSupported(const TensorInfo& input,
74 const TensorInfo& output,
75 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
76
77 bool IsFullyConnectedSupported(const TensorInfo& input,
78 const TensorInfo& output,
79 const TensorInfo& weights,
80 const TensorInfo& biases,
81 const FullyConnectedDescriptor& descriptor,
82 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
83
84 bool IsInputSupported(const TensorInfo& input,
85 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
86
87 bool IsL2NormalizationSupported(const TensorInfo& input,
88 const TensorInfo& output,
89 const L2NormalizationDescriptor& descriptor,
90 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
91
92 bool IsLstmSupported(const TensorInfo& input,
93 const TensorInfo& outputStateIn,
94 const TensorInfo& cellStateIn,
95 const TensorInfo& scratchBuffer,
96 const TensorInfo& outputStateOut,
97 const TensorInfo& cellStateOut,
98 const TensorInfo& output,
99 const LstmDescriptor& descriptor,
100 const TensorInfo& inputToForgetWeights,
101 const TensorInfo& inputToCellWeights,
102 const TensorInfo& inputToOutputWeights,
103 const TensorInfo& recurrentToForgetWeights,
104 const TensorInfo& recurrentToCellWeights,
105 const TensorInfo& recurrentToOutputWeights,
106 const TensorInfo& forgetGateBias,
107 const TensorInfo& cellBias,
108 const TensorInfo& outputGateBias,
109 const TensorInfo* inputToInputWeights,
110 const TensorInfo* recurrentToInputWeights,
111 const TensorInfo* cellToInputWeights,
112 const TensorInfo* inputGateBias,
113 const TensorInfo* projectionWeights,
114 const TensorInfo* projectionBias,
115 const TensorInfo* cellToForgetWeights,
116 const TensorInfo* cellToOutputWeights,
117 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
118
119 bool IsMeanSupported(const TensorInfo& input,
120 const TensorInfo& output,
121 const MeanDescriptor& descriptor,
122 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
123
124 bool IsMergerSupported(const std::vector<const TensorInfo*> inputs,
Nikhil Raj8599a412018-11-19 14:51:07 +0000[diff] [blame]125 const TensorInfo& output,
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]126 const OriginsDescriptor& descriptor,
127 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
128
129 bool IsMultiplicationSupported(const TensorInfo& input0,
130 const TensorInfo& input1,
131 const TensorInfo& output,
132 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
133
134 bool IsNormalizationSupported(const TensorInfo& input,
135 const TensorInfo& output,
136 const NormalizationDescriptor& descriptor,
137 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
138
139 bool IsOutputSupported(const TensorInfo& output,
140 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
141
142 bool IsPadSupported(const TensorInfo& input,
143 const TensorInfo& output,
144 const PadDescriptor& descriptor,
145 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
146
147 bool IsPermuteSupported(const TensorInfo& input,
148 const TensorInfo& output,
149 const PermuteDescriptor& descriptor,
150 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
151
152 bool IsPooling2dSupported(const TensorInfo& input,
153 const TensorInfo& output,
154 const Pooling2dDescriptor& descriptor,
155 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
156
157 bool IsReshapeSupported(const TensorInfo& input,
158 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
159
160 bool IsResizeBilinearSupported(const TensorInfo& input,
161 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
162
163 bool IsSoftmaxSupported(const TensorInfo& input,
164 const TensorInfo& output,
165 const SoftmaxDescriptor& descriptor,
166 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
167
168 bool IsSplitterSupported(const TensorInfo& input,
169 const ViewsDescriptor& descriptor,
170 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
171
172 bool IsSubtractionSupported(const TensorInfo& input0,
173 const TensorInfo& input1,
174 const TensorInfo& output,
175 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100[diff] [blame]176}; // class NeonLayerSupport
arovir014424b0a2018-10-04 10:46:04 +0100[diff] [blame]177
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]178} // namespace armnn