Blame - src/backends/neon/NeonLayerSupport.hpp - ml/armnn

telsoa01

4fcda01

2018-03-09 14:13:49 +0000

[diff] [blame]

1

//

2

David Beck

ecb56cd

2018-09-05 12:52:57 +0100

[diff] [blame]

3

// SPDX-License-Identifier: MIT

telsoa01

4fcda01

2018-03-09 14:13:49 +0000

[diff] [blame]

//

#pragma once

Matteo Martincigh

2019-01-10 17:34:20 +0000

[diff] [blame]

7

#include <backendsCommon/LayerSupportBase.hpp>

telsoa01

4fcda01

2018-03-09 14:13:49 +0000

[diff] [blame]

namespace armnn

{

Matteo Martincigh

2019-01-10 17:34:20 +0000

[diff] [blame]

12

class NeonLayerSupport : public LayerSupportBase

arovir01

4424b0a

2018-10-04 10:46:04 +0100

[diff] [blame]

13

{

arovir01

7ff76c5

2018-10-09 09:40:58 +0100

[diff] [blame]

14

public:

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

Jim Flynn

906f946

2019-05-10 13:55:21 +0100

[diff] [blame]

34

bool IsConcatSupported(const std::vector<const TensorInfo*> inputs,

35

const TensorInfo& output,

Jim Flynn

e242f2d

2019-05-22 14:24:13 +0100

[diff] [blame]

36

const ConcatDescriptor& descriptor,

Jim Flynn

906f946

2019-05-10 13:55:21 +0100

[diff] [blame]

37

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

38

arovir01

7ff76c5

2018-10-09 09:40:58 +0100

[diff] [blame]

39

bool IsConstantSupported(const TensorInfo& output,

40

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

41

42

bool IsConvertFp16ToFp32Supported(const TensorInfo& input,

43

const TensorInfo& output,

44

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

45

46

bool IsConvertFp32ToFp16Supported(const TensorInfo& input,

47

const TensorInfo& output,

48

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

49

50

bool IsConvolution2dSupported(const TensorInfo& input,

51

const TensorInfo& output,

52

const Convolution2dDescriptor& descriptor,

53

const TensorInfo& weights,

54

const Optional<TensorInfo>& biases,

55

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

56

57

bool IsDepthwiseConvolutionSupported(const TensorInfo& input,

58

const TensorInfo& output,

59

const DepthwiseConvolution2dDescriptor& descriptor,

60

const TensorInfo& weights,

61

const Optional<TensorInfo>& biases,

62

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

63

Narumol Prangnawarat

01961a7

2019-05-30 16:47:12 +0100

[diff] [blame^]

64

bool IsDequantizeSupported(const TensorInfo& input,

65

const TensorInfo& output,

66

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

67

Pablo Tello

f0bd683

2019-04-26 17:58:13 +0100

[diff] [blame]

68

bool IsDilatedDepthwiseConvolutionSupported(const TensorInfo& input,

69

const TensorInfo& output,

70

const DepthwiseConvolution2dDescriptor& descriptor,

71

const TensorInfo& weights,

72

const Optional<TensorInfo>& biases,

73

Optional<std::string&> reason = EmptyOptional()) const override;

74

arovir01

7ff76c5

2018-10-09 09:40:58 +0100

[diff] [blame]

75

bool IsFloorSupported(const TensorInfo& input,

76

const TensorInfo& output,

77

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

78

79

bool IsFullyConnectedSupported(const TensorInfo& input,

80

const TensorInfo& output,

81

const TensorInfo& weights,

82

const TensorInfo& biases,

83

const FullyConnectedDescriptor& descriptor,

84

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

85

kevmay01

eed8592

2019-01-28 08:37:25 +0000

[diff] [blame]

86

bool IsGreaterSupported(const TensorInfo& input0,

87

const TensorInfo& input1,

88

const TensorInfo& output,

89

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

90

arovir01

7ff76c5

2018-10-09 09:40:58 +0100

[diff] [blame]

91

bool IsInputSupported(const TensorInfo& input,

92

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

93

94

bool IsL2NormalizationSupported(const TensorInfo& input,

95

const TensorInfo& output,

96

const L2NormalizationDescriptor& descriptor,

97

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

98

Nattapat Chaimanowong

4e6597a

2018-12-20 14:14:06 +0000

[diff] [blame]

99

bool IsMaximumSupported(const TensorInfo& input0,

100

const TensorInfo& input1,

101

const TensorInfo& output,

102

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

103

arovir01

7ff76c5

2018-10-09 09:40:58 +0100

[diff] [blame]

104

bool IsMeanSupported(const TensorInfo& input,

105

const TensorInfo& output,

106

const MeanDescriptor& descriptor,

107

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

108

Matteo Martincigh

992d6dc

2019-01-10 17:34:20 +0000

[diff] [blame]

109

bool IsMemCopySupported(const TensorInfo& input,

110

const TensorInfo& output,

111

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

112

Jim Flynn

906f946

2019-05-10 13:55:21 +0100

[diff] [blame]

113

ARMNN_DEPRECATED_MSG("Use IsConcatSupported instead")

arovir01

7ff76c5

2018-10-09 09:40:58 +0100

[diff] [blame]

114

bool IsMergerSupported(const std::vector<const TensorInfo*> inputs,

Nikhil Raj

8599a41

2018-11-19 14:51:07 +0000

[diff] [blame]

115

const TensorInfo& output,

Jim Flynn

e242f2d

2019-05-22 14:24:13 +0100

[diff] [blame]

116

const MergerDescriptor& descriptor,

arovir01

7ff76c5

2018-10-09 09:40:58 +0100

[diff] [blame]

117

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

118

Conor Kennedy

54b2169

2019-01-09 07:57:38 +0000

[diff] [blame]

119

bool IsMinimumSupported(const TensorInfo& input0,

120

const TensorInfo& input1,

121

const TensorInfo& output,

122

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

123

arovir01

7ff76c5

2018-10-09 09:40:58 +0100

[diff] [blame]

124

bool IsMultiplicationSupported(const TensorInfo& input0,

125

const TensorInfo& input1,

126

const TensorInfo& output,

127

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

128

129

bool IsNormalizationSupported(const TensorInfo& input,

130

const TensorInfo& output,

131

const NormalizationDescriptor& descriptor,

132

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

133

134

bool IsOutputSupported(const TensorInfo& output,

135

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

136

Éanna Ó Catháin

1205574

2019-01-25 10:01:40 +0000

[diff] [blame]

137

bool IsPadSupported(const TensorInfo& input,

138

const TensorInfo& output,

139

const PadDescriptor& descriptor,

140

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

141

arovir01

7ff76c5

2018-10-09 09:40:58 +0100

[diff] [blame]

142

bool IsPermuteSupported(const TensorInfo& input,

143

const TensorInfo& output,

144

const PermuteDescriptor& descriptor,

145

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

146

147

bool IsPooling2dSupported(const TensorInfo& input,

148

const TensorInfo& output,

149

const Pooling2dDescriptor& descriptor,

150

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

151

Sadik Armagan

fabc289

2019-05-31 09:05:11 +0100

[diff] [blame]

152

bool IsQuantizeSupported(const TensorInfo& input,

153

const TensorInfo& output,

154

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

155

arovir01

7ff76c5

2018-10-09 09:40:58 +0100

[diff] [blame]

156

bool IsReshapeSupported(const TensorInfo& input,

Matteo Martincigh

992d6dc

2019-01-10 17:34:20 +0000

[diff] [blame]

157

const ReshapeDescriptor& descriptor,

arovir01

7ff76c5

2018-10-09 09:40:58 +0100

[diff] [blame]

158

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

159

Sadik Armagan

c625f00

2018-12-17 11:32:16 +0000

[diff] [blame]

160

bool IsResizeBilinearSupported(const TensorInfo& input,

161

const TensorInfo& output,

162

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

163

arovir01

7ff76c5

2018-10-09 09:40:58 +0100

[diff] [blame]

164

bool IsSoftmaxSupported(const TensorInfo& input,

165

const TensorInfo& output,

166

const SoftmaxDescriptor& descriptor,

167

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

168

Narumol Prangnawarat

15eb583

2019-05-20 15:31:05 +0100

[diff] [blame]

169

ARMNN_DEPRECATED_MSG("Use IsSplitterSupported with outputs instead")

arovir01

7ff76c5

2018-10-09 09:40:58 +0100

[diff] [blame]

170

bool IsSplitterSupported(const TensorInfo& input,

171

const ViewsDescriptor& descriptor,

172

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

173

Narumol Prangnawarat

15eb583

2019-05-20 15:31:05 +0100

[diff] [blame]

174

bool IsSplitterSupported(const TensorInfo& input,

175

const std::vector<std::reference_wrapper<TensorInfo>>& outputs,

176

const ViewsDescriptor& descriptor,

177

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

178

arovir01

7ff76c5

2018-10-09 09:40:58 +0100

[diff] [blame]

179

bool IsSubtractionSupported(const TensorInfo& input0,

180

const TensorInfo& input1,

181

const TensorInfo& output,

182

Optional<std::string&> reasonIfUnsupported = EmptyOptional()) const override;

Aron Virginas-Tar

fc82431

2018-10-15 15:00:13 +0100

[diff] [blame]

183

}; // class NeonLayerSupport

arovir01

4424b0a

2018-10-04 10:46:04 +0100

[diff] [blame]

184

arovir01

7ff76c5

2018-10-09 09:40:58 +0100

[diff] [blame]

185

} // namespace armnn