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(Cast, CastChefFactory)
  16 OP_CHEF(Ceil, CeilChefFactory)
  17 OP_CHEF(Concatenation, ConcatenationChefFactory)
  18 OP_CHEF(Conv2D, Conv2DChefFactory)
  19 OP_CHEF(Cos, CosChefFactory)
  20 OP_CHEF(DepthToSpace, DepthToSpaceChefFactory)
  21 OP_CHEF(DepthwiseConv2D, DepthwiseConv2DChefFactory)
  22 OP_CHEF(Div, DivChefFactory)
  23 OP_CHEF(ELU, ELUChefFactory)
  24 OP_CHEF(Equal, EqualChefFactory)
  25 OP_CHEF(Exp, ExpChefFactory)
  26 OP_CHEF(ExpandDims, ExpandDimsChefFactory)
  27 OP_CHEF(Fill, FillChefFactory)
  28 OP_CHEF(Floor, FloorChefFactory)
  29 OP_CHEF(FloorDiv, FloorDivChefFactory)
  30 OP_CHEF(FloorMod, FloorModChefFactory)
  31 OP_CHEF(FullyConnected, FullyConnectedChefFactory)
  32 OP_CHEF(Gather, GatherChefFactory)
  33 OP_CHEF(GatherNd, GatherNdChefFactory)
  34 OP_CHEF(Greater, GreaterChefFactory)
  35 OP_CHEF(GreaterEqual, GreaterEqualChefFactory)
  36 OP_CHEF(If, IfChefFactory)
  37 OP_CHEF(L2Normalize, L2NormalizeChefFactory)
  38 OP_CHEF(L2Pool2D, L2Pool2DChefFactory)
  39 OP_CHEF(LeakyRelu, LeakyReluChefFactory)
  40 OP_CHEF(Less, LessChefFactory)
  41 OP_CHEF(LessEqual, LessEqualChefFactory)
  42 OP_CHEF(LocalResponseNormalization, LocalResponseNormalizationChefFactory)
  43 OP_CHEF(Log, LogChefFactory)
  44 OP_CHEF(LogicalAnd, LogicalAndChefFactory)
  45 OP_CHEF(LogicalNot, LogicalNotChefFactory)
  46 OP_CHEF(LogicalOr, LogicalOrChefFactory)
  47 OP_CHEF(Logistic, LogisticChefFactory)
  48 OP_CHEF(LogSoftmax, LogSoftmaxChefFactory)
  49 OP_CHEF(MatrixDiag, MatrixDiagChefFactory)
  50 OP_CHEF(MatrixSetDiag, MatrixSetDiagChefFactory)
  51 OP_CHEF(Maximum, MaximumChefFactory)
  52 OP_CHEF(MaxPool2D, MaxPool2DChefFactory)
  53 OP_CHEF(Mean, MeanChefFactory)
  54 OP_CHEF(Minimum, MinimumChefFactory)
  55 OP_CHEF(MirrorPad, MirrorPadChefFactory)
  56 OP_CHEF(Mul, MulChefFactory)
  57 OP_CHEF(Neg, NegChefFactory)
  58 OP_CHEF(NonMaxSuppressionV4, NonMaxSuppressionV4ChefFactory)
  59 OP_CHEF(NotEqual, NotEqualChefFactory)
  60 OP_CHEF(OneHot, OneHotChefFactory)
  61 OP_CHEF(Pack, PackChefFactory)
  62 OP_CHEF(Pad, PadChefFactory)
  63 OP_CHEF(PadV2, PadV2ChefFactory)
  64 OP_CHEF(Pow, PowChefFactory)
  65 OP_CHEF(PRelu, PReluChefFactory)
  66 OP_CHEF(Range, RangeChefFactory)
  67 OP_CHEF(Rank, RankChefFactory)
  68 OP_CHEF(ReduceAny, ReduceAnyChefFactory)
  69 OP_CHEF(ReduceMax, ReduceMaxChefFactory)
  70 OP_CHEF(ReduceMin, ReduceMinChefFactory)
  71 OP_CHEF(ReduceProd, ReduceProdChefFactory)
  72 OP_CHEF(ReLU, ReLUChefFactory)
  73 OP_CHEF(ReLU6, ReLU6ChefFactory)
  74 OP_CHEF(ReLUN1To1, ReLUN1To1ChefFactory)
  75 OP_CHEF(Reshape, ReshapeChefFactory)
  76 OP_CHEF(ResizeBilinear, ResizeBilinearChefFactory)
  77 OP_CHEF(ResizeNearestNeighbor, ResizeNearestNeighborChefFactory)
  78 OP_CHEF(ReverseSequence, ReverseSequenceChefFactory)
  79 OP_CHEF(ReverseV2, ReverseV2ChefFactory)
  80 OP_CHEF(Round, RoundChefFactory)
  81 OP_CHEF(Rsqrt, RsqrtChefFactory)
  82 OP_CHEF(ScatterNd, ScatterNdChefFactory)
  83 OP_CHEF(SegmentSum,SegmentSumChefFactory)
  84 OP_CHEF(Select, SelectChefFactory)
  85 OP_CHEF(SelectV2, SelectV2ChefFactory)
  86 OP_CHEF(Shape, ShapeChefFactory)
  87 OP_CHEF(Sin, SinChefFactory)
  88 OP_CHEF(Slice, SliceChefFactory)
  89 OP_CHEF(Softmax, SoftmaxChefFactory)
  90 OP_CHEF(SpaceToBatchND, SpaceToBatchNDChefFactory)
  91 OP_CHEF(SpaceToDepth, SpaceToDepthChefFactory)
  92 OP_CHEF(SparseToDense, SparseToDenseChefFactory)
  93 OP_CHEF(Split, SplitChefFactory)
  94 OP_CHEF(SplitV, SplitVChefFactory)
  95 OP_CHEF(Sqrt, SqrtChefFactory)
  96 OP_CHEF(Square, SquareChefFactory)
  97 OP_CHEF(SquaredDifference, SquaredDifferenceChefFactory)
  98 OP_CHEF(Squeeze, SqueezeChefFactory)
  99 OP_CHEF(StridedSlice, StridedSliceChefFactory)
 100 OP_CHEF(Sub, SubChefFactory)
 101 OP_CHEF(Sum, SumChefFactory)
 102 OP_CHEF(Tanh, TanhChefFactory)
 103 OP_CHEF(Tile, TileChefFactory)
 104 OP_CHEF(TopKV2, TopKV2ChefFactory)
 105 OP_CHEF(Transpose, TransposeChefFactory)
 106 OP_CHEF(TransposeConv, TransposeConvChefFactory)
 107 OP_CHEF(Unique, UniqueChefFactory)
 108 OP_CHEF(Unpack, UnpackChefFactory)
 109 OP_CHEF(Where, WhereChefFactory)
 110 OP_CHEF(While, WhileChefFactory)
 111 OP_CHEF(ZerosLike, ZerosLikeChefFactory)
 112
 113 // Custom Op
 114 OP_CHEF(AddV2, AddV2ChefFactory)
 115 OP_CHEF(All, AllChefFactory)
 116 OP_CHEF(BatchMatMulV2, BatchMatMulV2ChefFactory)
 117 OP_CHEF(MatMul, MatMulChefFactory)
 118 OP_CHEF(MatrixBandPart, MatrixBandPartChefFactory)