blob: 85bb5c9346a20959cce2d14b4d8822b99bf15965 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
// Copyright (c) 2020-2021, ARM Limited.
//
// Licensed under the Apache License, Version 2.0 with LLVM Exceptions
// (the "License"); you may not use this file except in compliance with
// the License. You may obtain a copy of the License at
//
// https://llvm.org/LICENSE.txt
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
/*
Syntax:
DEF_OPERATOR(MLIR_OP)
Description:
MLIR_OP: the operator class type, must match mlir/include/mlir/Dialect/Tosa/IR/tosa_ops.td in llvm-project repo
*/
/* tensor operators */
DEF_OPERATOR(ArgMax)
DEF_OPERATOR(AvgPool2d)
DEF_OPERATOR(Conv2D)
DEF_OPERATOR(Conv3D)
DEF_OPERATOR(DepthwiseConv2D)
DEF_OPERATOR(FullyConnected)
DEF_OPERATOR(MatMul)
DEF_OPERATOR(MaxPool2d)
DEF_OPERATOR(TransposeConv2D)
/* activation */
DEF_OPERATOR(Clamp)
DEF_OPERATOR(Sigmoid)
DEF_OPERATOR(Tanh)
/* elementwise - binary */
DEF_OPERATOR(Add)
DEF_OPERATOR(ArithmeticRightShift)
DEF_OPERATOR(BitwiseAnd)
DEF_OPERATOR(BitwiseOr)
DEF_OPERATOR(BitwiseXor)
DEF_OPERATOR(Div)
DEF_OPERATOR(LogicalAnd)
DEF_OPERATOR(LogicalLeftShift)
DEF_OPERATOR(LogicalRightShift)
DEF_OPERATOR(LogicalOr)
DEF_OPERATOR(LogicalXor)
DEF_OPERATOR(Maximum)
DEF_OPERATOR(Minimum)
DEF_OPERATOR(Mul)
DEF_OPERATOR(Pow)
DEF_OPERATOR(Sub)
DEF_OPERATOR(Table)
/* elementwise - unary */
DEF_OPERATOR(Abs)
DEF_OPERATOR(BitwiseNot)
DEF_OPERATOR(Ceil)
DEF_OPERATOR(Clz)
DEF_OPERATOR(Exp)
DEF_OPERATOR(Floor)
DEF_OPERATOR(Log)
DEF_OPERATOR(LogicalNot)
DEF_OPERATOR(Negate)
DEF_OPERATOR(Reciprocal)
DEF_OPERATOR(Rsqrt)
/* elementwise - ternary */
DEF_OPERATOR(Select)
/* logical */
DEF_OPERATOR(Equal)
DEF_OPERATOR(Greater)
DEF_OPERATOR(GreaterEqual)
/* reduction */
DEF_OPERATOR(ReduceAny)
DEF_OPERATOR(ReduceAll)
DEF_OPERATOR(ReduceMax)
DEF_OPERATOR(ReduceMin)
DEF_OPERATOR(ReduceProd)
DEF_OPERATOR(ReduceSum)
/* memory operation */
DEF_OPERATOR(Concat)
DEF_OPERATOR(Pad)
DEF_OPERATOR(Reshape)
DEF_OPERATOR(Reverse)
DEF_OPERATOR(Slice)
DEF_OPERATOR(Tile)
DEF_OPERATOR(Transpose)
/* gather/scatter */
DEF_OPERATOR(Gather)
DEF_OPERATOR(Scatter)
/* image */
DEF_OPERATOR(Resize)
/* quantization */
DEF_OPERATOR(Cast)
DEF_OPERATOR(Rescale)
/* data nodes */
DEF_OPERATOR(Const)
DEF_OPERATOR(Identity)
/* custom operations */
DEF_OPERATOR(Custom)
/* control flow operators */
DEF_OPERATOR(If)
DEF_OPERATOR(While)
|
