compiler/tflchef/core/src/OpChef.def

   1 #ifndef OP_CHEF
   2 #error "Define OP first"
   3 #endif // OP_CHEF
   4
   5 // Please keep the list in alphabetical order
   6 // OP_CHEF(NAME, FACTORY_CLASS)
   7 OP_CHEF(Abs, AbsChefFactory)
   8 OP_CHEF(Add, AddChefFactory)
   9 OP_CHEF(AddN, AddNChefFactory)
  10 OP_CHEF(ArgMax, ArgMaxChefFactory)
  11 OP_CHEF(ArgMin, ArgMinChefFactory)
  12 OP_CHEF(AveragePool2D, AveragePool2DChefFactory)
  13 OP_CHEF(BatchMatMul, BatchMatMulChefFactory)
  14 OP_CHEF(BatchToSpaceND, BatchToSpaceNDChefFactory)
  15 OP_CHEF(BidirectionalSequenceLSTM, BidirectionalSequenceLSTMChefFactory)
  16 OP_CHEF(Cast, CastChefFactory)
  17 OP_CHEF(Ceil, CeilChefFactory)
  18 OP_CHEF(Concatenation, ConcatenationChefFactory)
  19 OP_CHEF(Conv2D, Conv2DChefFactory)
  20 OP_CHEF(Cos, CosChefFactory)
  21 OP_CHEF(Densify, DensifyChefFactory)
  22 OP_CHEF(DepthToSpace, DepthToSpaceChefFactory)
  23 OP_CHEF(DepthwiseConv2D, DepthwiseConv2DChefFactory)
  24 OP_CHEF(Dequantize, DequantizeChefFactory)
  25 OP_CHEF(Div, DivChefFactory)
  26 OP_CHEF(ELU, ELUChefFactory)
  27 OP_CHEF(Equal, EqualChefFactory)
  28 OP_CHEF(Exp, ExpChefFactory)
  29 OP_CHEF(ExpandDims, ExpandDimsChefFactory)
  30 OP_CHEF(FakeQuant, FakeQuantChefFactory)
  31 OP_CHEF(Fill, FillChefFactory)
  32 OP_CHEF(Floor, FloorChefFactory)
  33 OP_CHEF(FloorDiv, FloorDivChefFactory)
  34 OP_CHEF(FloorMod, FloorModChefFactory)
  35 OP_CHEF(FullyConnected, FullyConnectedChefFactory)
  36 OP_CHEF(Gather, GatherChefFactory)
  37 OP_CHEF(GatherNd, GatherNdChefFactory)
  38 OP_CHEF(Gelu, GeluChefFactory)
  39 OP_CHEF(Greater, GreaterChefFactory)
  40 OP_CHEF(GreaterEqual, GreaterEqualChefFactory)
  41 OP_CHEF(HardSwish, HardSwishChefFactory)
  42 OP_CHEF(If, IfChefFactory)
  43 OP_CHEF(L2Normalize, L2NormalizeChefFactory)
  44 OP_CHEF(L2Pool2D, L2Pool2DChefFactory)
  45 OP_CHEF(LeakyRelu, LeakyReluChefFactory)
  46 OP_CHEF(Less, LessChefFactory)
  47 OP_CHEF(LessEqual, LessEqualChefFactory)
  48 OP_CHEF(LocalResponseNormalization, LocalResponseNormalizationChefFactory)
  49 OP_CHEF(Log, LogChefFactory)
  50 OP_CHEF(LogicalAnd, LogicalAndChefFactory)
  51 OP_CHEF(LogicalNot, LogicalNotChefFactory)
  52 OP_CHEF(LogicalOr, LogicalOrChefFactory)
  53 OP_CHEF(Logistic, LogisticChefFactory)
  54 OP_CHEF(LogSoftmax, LogSoftmaxChefFactory)
  55 OP_CHEF(MatrixDiag, MatrixDiagChefFactory)
  56 OP_CHEF(MatrixSetDiag, MatrixSetDiagChefFactory)
  57 OP_CHEF(Maximum, MaximumChefFactory)
  58 OP_CHEF(MaxPool2D, MaxPool2DChefFactory)
  59 OP_CHEF(Mean, MeanChefFactory)
  60 OP_CHEF(Minimum, MinimumChefFactory)
  61 OP_CHEF(MirrorPad, MirrorPadChefFactory)
  62 OP_CHEF(Mul, MulChefFactory)
  63 OP_CHEF(Neg, NegChefFactory)
  64 OP_CHEF(NonMaxSuppressionV4, NonMaxSuppressionV4ChefFactory)
  65 OP_CHEF(NonMaxSuppressionV5, NonMaxSuppressionV5ChefFactory)
  66 OP_CHEF(NotEqual, NotEqualChefFactory)
  67 OP_CHEF(OneHot, OneHotChefFactory)
  68 OP_CHEF(Pack, PackChefFactory)
  69 OP_CHEF(Pad, PadChefFactory)
  70 OP_CHEF(PadV2, PadV2ChefFactory)
  71 OP_CHEF(Pow, PowChefFactory)
  72 OP_CHEF(PRelu, PReluChefFactory)
  73 OP_CHEF(Quantize, QuantizeChefFactory)
  74 OP_CHEF(Range, RangeChefFactory)
  75 OP_CHEF(Rank, RankChefFactory)
  76 OP_CHEF(ReduceAny, ReduceAnyChefFactory)
  77 OP_CHEF(ReduceMax, ReduceMaxChefFactory)
  78 OP_CHEF(ReduceMin, ReduceMinChefFactory)
  79 OP_CHEF(ReduceProd, ReduceProdChefFactory)
  80 OP_CHEF(ReLU, ReLUChefFactory)
  81 OP_CHEF(ReLU6, ReLU6ChefFactory)
  82 OP_CHEF(ReLUN1To1, ReLUN1To1ChefFactory)
  83 OP_CHEF(Reshape, ReshapeChefFactory)
  84 OP_CHEF(ResizeBilinear, ResizeBilinearChefFactory)
  85 OP_CHEF(ResizeNearestNeighbor, ResizeNearestNeighborChefFactory)
  86 OP_CHEF(ReverseSequence, ReverseSequenceChefFactory)
  87 OP_CHEF(ReverseV2, ReverseV2ChefFactory)
  88 OP_CHEF(Round, RoundChefFactory)
  89 OP_CHEF(Rsqrt, RsqrtChefFactory)
  90 OP_CHEF(ScatterNd, ScatterNdChefFactory)
  91 OP_CHEF(SegmentSum,SegmentSumChefFactory)
  92 OP_CHEF(Select, SelectChefFactory)
  93 OP_CHEF(SelectV2, SelectV2ChefFactory)
  94 OP_CHEF(Shape, ShapeChefFactory)
  95 OP_CHEF(Sin, SinChefFactory)
  96 OP_CHEF(Slice, SliceChefFactory)
  97 OP_CHEF(Softmax, SoftmaxChefFactory)
  98 OP_CHEF(SpaceToBatchND, SpaceToBatchNDChefFactory)
  99 OP_CHEF(SpaceToDepth, SpaceToDepthChefFactory)
 100 OP_CHEF(SparseToDense, SparseToDenseChefFactory)
 101 OP_CHEF(Split, SplitChefFactory)
 102 OP_CHEF(SplitV, SplitVChefFactory)
 103 OP_CHEF(Sqrt, SqrtChefFactory)
 104 OP_CHEF(Square, SquareChefFactory)
 105 OP_CHEF(SquaredDifference, SquaredDifferenceChefFactory)
 106 OP_CHEF(Squeeze, SqueezeChefFactory)
 107 OP_CHEF(StridedSlice, StridedSliceChefFactory)
 108 OP_CHEF(Sub, SubChefFactory)
 109 OP_CHEF(Sum, SumChefFactory)
 110 OP_CHEF(SVDF, SVDFChefFactory)
 111 OP_CHEF(Tanh, TanhChefFactory)
 112 OP_CHEF(Tile, TileChefFactory)
 113 OP_CHEF(TopKV2, TopKV2ChefFactory)
 114 OP_CHEF(Transpose, TransposeChefFactory)
 115 OP_CHEF(TransposeConv, TransposeConvChefFactory)
 116 OP_CHEF(UnidirectionalSequenceLSTM, UnidirectionalSequenceLSTMChefFactory)
 117 OP_CHEF(Unique, UniqueChefFactory)
 118 OP_CHEF(Unpack, UnpackChefFactory)
 119 OP_CHEF(Where, WhereChefFactory)
 120 OP_CHEF(While, WhileChefFactory)
 121 OP_CHEF(ZerosLike, ZerosLikeChefFactory)
 122
 123 // Custom Op
 124 OP_CHEF(AddV2, AddV2ChefFactory)
 125 OP_CHEF(All, AllChefFactory)
 126 OP_CHEF(BatchMatMulV2, BatchMatMulV2ChefFactory)
 127 OP_CHEF(BroadcastTo, BroadcastToChefFactory)
 128 OP_CHEF(Erf, ErfChefFactory)
 129 OP_CHEF(MatMul, MatMulChefFactory)
 130 OP_CHEF(MatrixBandPart, MatrixBandPartChefFactory)
 131 OP_CHEF(MaxPoolWithArgmax, MaxPoolWithArgmaxChefFactory)