Gitiles
Code Review Sign In
review.mlplatform.org / ml / armnn / 861985ff2964720a0165e109c3fc568cb245bbe9 / . / src / backends / neon / NeonLayerSupport.hpp
blob: 27e825afa9acf34643b88b316b43390d9fa0e2bc [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
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]7#include <backendsCommon/LayerSupportBase.hpp>
telsoa014fcda012018-03-09 14:13:49 +0000[diff] [blame]8
9namespace armnn
10{
11
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]12class NeonLayerSupport : public LayerSupportBase
arovir014424b0a2018-10-04 10:46:04 +0100[diff] [blame]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
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]59 bool IsFloorSupported(const TensorInfo& input,
60 const TensorInfo& output,
61 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
62
63 bool IsFullyConnectedSupported(const TensorInfo& input,
64 const TensorInfo& output,
65 const TensorInfo& weights,
66 const TensorInfo& biases,
67 const FullyConnectedDescriptor& descriptor,
68 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
69
kevmay01eed85922019-01-28 08:37:25 +0000[diff] [blame]70 bool IsGreaterSupported(const TensorInfo& input0,
71 const TensorInfo& input1,
72 const TensorInfo& output,
73 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
74
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]75 bool IsInputSupported(const TensorInfo& input,
76 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
77
78 bool IsL2NormalizationSupported(const TensorInfo& input,
79 const TensorInfo& output,
80 const L2NormalizationDescriptor& descriptor,
81 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
82
Nattapat Chaimanowong4e6597a2018-12-20 14:14:06 +0000[diff] [blame]83 bool IsMaximumSupported(const TensorInfo& input0,
84 const TensorInfo& input1,
85 const TensorInfo& output,
86 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
87
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]88 bool IsMeanSupported(const TensorInfo& input,
89 const TensorInfo& output,
90 const MeanDescriptor& descriptor,
91 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
92
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]93 bool IsMemCopySupported(const TensorInfo& input,
94 const TensorInfo& output,
95 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
96
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]97 bool IsMergerSupported(const std::vector<const TensorInfo*> inputs,
Nikhil Raj8599a412018-11-19 14:51:07 +0000[diff] [blame]98 const TensorInfo& output,
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]99 const OriginsDescriptor& descriptor,
100 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
101
Conor Kennedy54b21692019-01-09 07:57:38 +0000[diff] [blame]102 bool IsMinimumSupported(const TensorInfo& input0,
103 const TensorInfo& input1,
104 const TensorInfo& output,
105 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
106
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]107 bool IsMultiplicationSupported(const TensorInfo& input0,
108 const TensorInfo& input1,
109 const TensorInfo& output,
110 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
111
112 bool IsNormalizationSupported(const TensorInfo& input,
113 const TensorInfo& output,
114 const NormalizationDescriptor& descriptor,
115 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
116
117 bool IsOutputSupported(const TensorInfo& output,
118 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
119
Éanna Ó Catháin12055742019-01-25 10:01:40 +0000[diff] [blame]120 bool IsPadSupported(const TensorInfo& input,
121 const TensorInfo& output,
122 const PadDescriptor& descriptor,
123 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
124
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]125 bool IsPermuteSupported(const TensorInfo& input,
126 const TensorInfo& output,
127 const PermuteDescriptor& descriptor,
128 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
129
130 bool IsPooling2dSupported(const TensorInfo& input,
131 const TensorInfo& output,
132 const Pooling2dDescriptor& descriptor,
133 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
134
135 bool IsReshapeSupported(const TensorInfo& input,
Matteo Martincigh992d6dc2019-01-10 17:34:20 +0000[diff] [blame]136 const ReshapeDescriptor& descriptor,
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]137 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
138
Sadik Armaganc625f002018-12-17 11:32:16 +0000[diff] [blame]139 bool IsResizeBilinearSupported(const TensorInfo& input,
140 const TensorInfo& output,
141 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
142
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]143 bool IsSoftmaxSupported(const TensorInfo& input,
144 const TensorInfo& output,
145 const SoftmaxDescriptor& descriptor,
146 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
147
148 bool IsSplitterSupported(const TensorInfo& input,
149 const ViewsDescriptor& descriptor,
150 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
151
152 bool IsSubtractionSupported(const TensorInfo& input0,
153 const TensorInfo& input1,
154 const TensorInfo& output,
155 Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;
Aron Virginas-Tarfc824312018-10-15 15:00:13 +0100[diff] [blame]156}; // class NeonLayerSupport
arovir014424b0a2018-10-04 10:46:04 +0100[diff] [blame]157
arovir017ff76c52018-10-09 09:40:58 +0100[diff] [blame]158} // namespace armnn