summaryrefslogtreecommitdiff
path: root/compiler/tflchef/core/src/OpChef.def
blob: a25250c46dd5de808ab50d023419064e1fd1a702 (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
#ifndef OP_CHEF
#error "Define OP first"
#endif // OP_CHEF

// Please keep the list in alphabetical order
// OP_CHEF(NAME, FACTORY_CLASS)
OP_CHEF(Abs, AbsChefFactory)
OP_CHEF(Add, AddChefFactory)
OP_CHEF(ArgMax, ArgMaxChefFactory)
OP_CHEF(BatchToSpaceND, BatchToSpaceNDChefFactory)
OP_CHEF(AveragePool2D, AveragePool2DChefFactory)
OP_CHEF(Concatenation, ConcatenationChefFactory)
OP_CHEF(Conv2D, Conv2DChefFactory)
OP_CHEF(Cos, CosChefFactory)
OP_CHEF(DepthwiseConv2D, DepthwiseConv2DChefFactory)
OP_CHEF(Div, DivChefFactory)
OP_CHEF(Equal, EqualChefFactory)
OP_CHEF(Exp, ExpChefFactory)
OP_CHEF(FloorDiv, FloorDivChefFactory)
OP_CHEF(FullyConnected, FullyConnectedChefFactory)
OP_CHEF(LogicalNot, LogicalNotChefFactory)
OP_CHEF(LogicalOr, LogicalOrChefFactory)
OP_CHEF(MaxPool2D, MaxPool2DChefFactory)
OP_CHEF(Mean, MeanChefFactory)
OP_CHEF(Mul, MulChefFactory)
OP_CHEF(Pack, PackChefFactory)
OP_CHEF(Pad, PadChefFactory)
OP_CHEF(ReLU, ReLUChefFactory)
OP_CHEF(ReLU6, ReLU6ChefFactory)
OP_CHEF(Reshape, ReshapeChefFactory)
OP_CHEF(Rsqrt, RsqrtChefFactory)
OP_CHEF(Shape, ShapeChefFactory)
OP_CHEF(Softmax, SoftmaxChefFactory)
OP_CHEF(Sqrt, SqrtChefFactory)
OP_CHEF(Sub, SubChefFactory)
OP_CHEF(Tanh, TanhChefFactory)
OP_CHEF(Transpose, TransposeChefFactory)