ops_supported.inc 470 Bytes
Newer Older
1 2 3 4 5 6
// List of all ops supported by MLIR backend end-to-end
#ifndef MLIR_OP
#define MLIR_OP
#endif

MLIR_OP(Add)
7
MLIR_OP(ArgMin)
nmostafa's avatar
nmostafa committed
8
MLIR_OP(ArgMax)
9
MLIR_OP(Divide)
10
MLIR_OP(Dot)
Adam Procter's avatar
Adam Procter committed
11
MLIR_OP(Concat)
12
MLIR_OP(Convolution)
nmostafa's avatar
nmostafa committed
13
MLIR_OP(Gather)
14 15
MLIR_OP(Greater)
MLIR_OP(Less)
16 17 18 19
MLIR_OP(GreaterEq)
MLIR_OP(LessEq)
MLIR_OP(Equal)
MLIR_OP(NotEqual)
20 21 22
MLIR_OP(Maximum)
MLIR_OP(Minimum)
MLIR_OP(Multiply)
23
MLIR_OP(Negative)
24 25
MLIR_OP(Subtract)
MLIR_OP(Relu)
26 27
// Add new supported ops here

nmostafa's avatar
nmostafa committed
28
#undef MLIR_OP