compiler/luci/lang/include/luci/IR/CircleNodes.lst

   1 #ifndef CIRCLE_NODE
   2 #error "Define CIRCLE_NODE"
   3 #endif // CIRCLE_NODE
   4
   5 //
   6 // PLEASE SORT NODE DECLS IN ALPHABETICAL ORDER
   7 //
   8 // Naming rule: Follow names in TensorFlow C++ source; same as TFDialect
   9 // ex) for AvgPool, tensorflow/core/ops/nn_ops.cc
  10 //    REGISTER_OP("AvgPool")     <-- OPCODE: AvgPool. Prefix `Circle` for CLASS name
  11 //        .Input("value: T")     <-- Input name is 'value'
  12 //
  13
  14 CIRCLE_NODE(ABS, luci::CircleAbs)
  15 CIRCLE_NODE(ADD, luci::CircleAdd)
  16 CIRCLE_NODE(ADD_N, luci::CircleAddN)
  17 CIRCLE_NODE(ARG_MAX, luci::CircleArgMax)
  18 CIRCLE_NODE(ARG_MIN, luci::CircleArgMin)
  19 CIRCLE_NODE(AVERAGE_POOL_2D, luci::CircleAveragePool2D)
  20 CIRCLE_NODE(BATCH_TO_SPACE_ND, luci::CircleBatchToSpaceND)
  21 CIRCLE_NODE(BATCHMATMUL, luci::CircleBatchMatMul)
  22 CIRCLE_NODE(CAST, luci::CircleCast)
  23 CIRCLE_NODE(CEIL, luci::CircleCeil)
  24 CIRCLE_NODE(CONCATENATION, luci::CircleConcatenation)
  25 CIRCLE_NODE(CONV_2D, luci::CircleConv2D)
  26 CIRCLE_NODE(COS, luci::CircleCos)
  27 CIRCLE_NODE(CUSTOM, luci::CircleCustom)
  28 CIRCLE_NODE(DEPTH_TO_SPACE, luci::CircleDepthToSpace)
  29 CIRCLE_NODE(DEPTHWISE_CONV_2D, luci::CircleDepthwiseConv2D)
  30 CIRCLE_NODE(DIV, luci::CircleDiv)
  31 CIRCLE_NODE(ELU, luci::CircleElu)
  32 CIRCLE_NODE(EQUAL, luci::CircleEqual)
  33 CIRCLE_NODE(EXP, luci::CircleExp)
  34 CIRCLE_NODE(EXPAND_DIMS, luci::CircleExpandDims)
  35 CIRCLE_NODE(FILL, luci::CircleFill)
  36 CIRCLE_NODE(FLOOR, luci::CircleFloor)
  37 CIRCLE_NODE(FLOOR_DIV, luci::CircleFloorDiv)
  38 CIRCLE_NODE(FLOOR_MOD, luci::CircleFloorMod)
  39 CIRCLE_NODE(FULLY_CONNECTED, luci::CircleFullyConnected)
  40 CIRCLE_NODE(GATHER, luci::CircleGather)
  41 CIRCLE_NODE(GATHER_ND, luci::CircleGatherNd)
  42 CIRCLE_NODE(GREATER, luci::CircleGreater)
  43 CIRCLE_NODE(GREATER_EQUAL, luci::CircleGreaterEqual)
  44 CIRCLE_NODE(IF, luci::CircleIf)
  45 CIRCLE_NODE(L2_NORMALIZATION, luci::CircleL2Normalize)
  46 CIRCLE_NODE(L2_POOL_2D, luci::CircleL2Pool2D)
  47 CIRCLE_NODE(LEAKY_RELU, luci::CircleLeakyRelu)
  48 CIRCLE_NODE(LESS, luci::CircleLess)
  49 CIRCLE_NODE(LESS_EQUAL, luci::CircleLessEqual)
  50 CIRCLE_NODE(LOCAL_RESPONSE_NORMALIZATION, luci::CircleLocalResponseNormalization)
  51 CIRCLE_NODE(LOG, luci::CircleLog)
  52 CIRCLE_NODE(LOGICAL_AND, luci::CircleLogicalAnd)
  53 CIRCLE_NODE(LOGICAL_NOT, luci::CircleLogicalNot)
  54 CIRCLE_NODE(LOGICAL_OR, luci::CircleLogicalOr)
  55 CIRCLE_NODE(LOGISTIC, luci::CircleLogistic)
  56 CIRCLE_NODE(LOG_SOFTMAX, luci::CircleLogSoftmax)
  57 CIRCLE_NODE(MATRIX_DIAG, luci::CircleMatrixDiag)
  58 CIRCLE_NODE(MAX_POOL_2D, luci::CircleMaxPool2D)
  59 CIRCLE_NODE(MATRIX_SET_DIAG, luci::CircleMatrixSetDiag)
  60 CIRCLE_NODE(MAXIMUM, luci::CircleMaximum)
  61 CIRCLE_NODE(MEAN, luci::CircleMean)
  62 CIRCLE_NODE(MINIMUM, luci::CircleMinimum)
  63 CIRCLE_NODE(MIRROR_PAD, luci::CircleMirrorPad)
  64 CIRCLE_NODE(MUL, luci::CircleMul)
  65 CIRCLE_NODE(NEG, luci::CircleNeg)
  66 CIRCLE_NODE(NON_MAX_SUPPRESSION_V4, luci::CircleNonMaxSuppressionV4)
  67 CIRCLE_NODE(NOT_EQUAL, luci::CircleNotEqual)
  68 CIRCLE_NODE(ONE_HOT, luci::CircleOneHot)
  69 CIRCLE_NODE(PACK, luci::CirclePack)
  70 CIRCLE_NODE(PAD, luci::CirclePad)
  71 CIRCLE_NODE(PADV2, luci::CirclePadV2)
  72 CIRCLE_NODE(POW, luci::CirclePow)
  73 CIRCLE_NODE(PRELU, luci::CirclePRelu)
  74 CIRCLE_NODE(RANGE, luci::CircleRange)
  75 CIRCLE_NODE(RANK, luci::CircleRank)
  76 CIRCLE_NODE(REDUCE_ANY, luci::CircleReduceAny)
  77 CIRCLE_NODE(REDUCE_MAX, luci::CircleReduceMax)
  78 CIRCLE_NODE(REDUCE_MIN, luci::CircleReduceMin)
  79 CIRCLE_NODE(REDUCE_PROD, luci::CircleReduceProd)
  80 CIRCLE_NODE(RELU, luci::CircleRelu)
  81 CIRCLE_NODE(RELU6, luci::CircleRelu6)
  82 CIRCLE_NODE(RELU_N1_TO_1, luci::CircleReluN1To1)
  83 CIRCLE_NODE(RESHAPE, luci::CircleReshape)
  84 CIRCLE_NODE(RESIZE_BILINEAR, luci::CircleResizeBilinear)
  85 CIRCLE_NODE(RESIZE_NEAREST_NEIGHBOR, luci::CircleResizeNearestNeighbor)
  86 CIRCLE_NODE(REVERSE_SEQUENCE, luci::CircleReverseSequence)
  87 CIRCLE_NODE(REVERSE_V2, luci::CircleReverseV2)
  88 CIRCLE_NODE(ROUND, luci::CircleRound)
  89 CIRCLE_NODE(RSQRT, luci::CircleRsqrt)
  90 CIRCLE_NODE(SCATTER_ND, luci::CircleScatterNd)
  91 CIRCLE_NODE(SEGMENT_SUM, luci::CircleSegmentSum)
  92 CIRCLE_NODE(SELECT, luci::CircleSelect)
  93 CIRCLE_NODE(SELECT_V2, luci::CircleSelectV2)
  94 CIRCLE_NODE(SHAPE, luci::CircleShape)
  95 CIRCLE_NODE(SIN, luci::CircleSin)
  96 CIRCLE_NODE(SLICE, luci::CircleSlice)
  97 CIRCLE_NODE(SOFTMAX, luci::CircleSoftmax)
  98 CIRCLE_NODE(SPACE_TO_BATCH_ND, luci::CircleSpaceToBatchND)
  99 CIRCLE_NODE(SPACE_TO_DEPTH, luci::CircleSpaceToDepth)
 100 CIRCLE_NODE(SPARSE_TO_DENSE, luci::CircleSparseToDense)
 101 CIRCLE_NODE(SPLIT, luci::CircleSplit)
 102 CIRCLE_NODE(SPLIT_V, luci::CircleSplitV)
 103 CIRCLE_NODE(SQRT, luci::CircleSqrt)
 104 CIRCLE_NODE(SQUARE, luci::CircleSquare)
 105 CIRCLE_NODE(SQUARED_DIFFERENCE, luci::CircleSquaredDifference)
 106 CIRCLE_NODE(SQUEEZE, luci::CircleSqueeze)
 107 CIRCLE_NODE(STRIDED_SLICE, luci::CircleStridedSlice)
 108 CIRCLE_NODE(SUB, luci::CircleSub)
 109 CIRCLE_NODE(SUM, luci::CircleSum)
 110 CIRCLE_NODE(TANH, luci::CircleTanh)
 111 CIRCLE_NODE(TILE, luci::CircleTile)
 112 CIRCLE_NODE(TOPK_V2, luci::CircleTopKV2)
 113 CIRCLE_NODE(TRANSPOSE, luci::CircleTranspose)
 114 CIRCLE_NODE(TRANSPOSE_CONV, luci::CircleTransposeConv)
 115 CIRCLE_NODE(UNIQUE, luci::CircleUnique)
 116 CIRCLE_NODE(UNPACK, luci::CircleUnpack)
 117 CIRCLE_NODE(WHERE, luci::CircleWhere)
 118 CIRCLE_NODE(WHILE, luci::CircleWhile)
 119 CIRCLE_NODE(ZEROS_LIKE, luci::CircleZerosLike)
 120 // Circle Only
 121 CIRCLE_NODE(BCQ_FULLY_CONNECTED, luci::CircleBCQFullyConnected)
 122 CIRCLE_NODE(BCQ_GATHER, luci::CircleBCQGather)
 123 CIRCLE_NODE(INSTANCE_NORM, luci::CircleInstanceNorm)
 124 // Virtual node(s)
 125 CIRCLE_NODE(CIRCLECONST, luci::CircleConst)
 126 CIRCLE_NODE(CIRCLEINPUT, luci::CircleInput)
 127 CIRCLE_NODE(CIRCLEOUTPUT, luci::CircleOutput)
 128 CIRCLE_NODE(CIRCLEOUTPUTDUMMY, luci::CircleOutputDummy)
 129 CIRCLE_NODE(CIRCLEOUTPUTEXCLUDE, luci::CircleOutputExclude)
 130 CIRCLE_NODE(CIRCLECUSTOMOUT, luci::CircleCustomOut)
 131 CIRCLE_NODE(CIRCLEIFOUT, luci::CircleIfOut)
 132 CIRCLE_NODE(CIRCLENONMAXSUPPRESSIONV4OUT, luci::CircleNonMaxSuppressionV4Out)
 133 CIRCLE_NODE(CIRCLESPLITOUT, luci::CircleSplitOut)
 134 CIRCLE_NODE(CIRCLESPLITVOUT, luci::CircleSplitVOut)
 135 CIRCLE_NODE(CIRCLETOPKV2OUT, luci::CircleTopKV2Out)
 136 CIRCLE_NODE(CIRCLEUNIQUEOUT, luci::CircleUniqueOut)
 137 CIRCLE_NODE(CIRCLEUNPACKOUT, luci::CircleUnpackOut)
 138 CIRCLE_NODE(CIRCLEWHILEOUT, luci::CircleWhileOut)