summaryrefslogtreecommitdiff
path: root/compiler/luci/lang/include/luci/IR/CircleNodes.lst
blob: 9f0a1b16e62935f4ee09d26b07a8a92eef8e1c75 (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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
#ifndef CIRCLE_NODE
#error "Define CIRCLE_NODE"
#endif // CIRCLE_NODE

//
// PLEASE SORT NODE DECLS IN ALPHABETICAL ORDER
//
// Naming rule: Follow names in TensorFlow C++ source; same as TFDialect
// ex) for AvgPool, tensorflow/core/ops/nn_ops.cc
//    REGISTER_OP("AvgPool")     <-- OPCODE: AvgPool. Prefix `Circle` for CLASS name
//        .Input("value: T")     <-- Input name is 'value'
//

CIRCLE_NODE(ABS, luci::CircleAbs)
CIRCLE_NODE(ADD, luci::CircleAdd)
CIRCLE_NODE(ADD_N, luci::CircleAddN)
CIRCLE_NODE(ARG_MAX, luci::CircleArgMax)
CIRCLE_NODE(ARG_MIN, luci::CircleArgMin)
CIRCLE_NODE(AVERAGE_POOL_2D, luci::CircleAveragePool2D)
CIRCLE_NODE(BATCH_TO_SPACE_ND, luci::CircleBatchToSpaceND)
CIRCLE_NODE(BATCHMATMUL, luci::CircleBatchMatMul)
CIRCLE_NODE(CAST, luci::CircleCast)
CIRCLE_NODE(CEIL, luci::CircleCeil)
CIRCLE_NODE(CONCATENATION, luci::CircleConcatenation)
CIRCLE_NODE(CONV_2D, luci::CircleConv2D)
CIRCLE_NODE(COS, luci::CircleCos)
CIRCLE_NODE(CUSTOM, luci::CircleCustom)
CIRCLE_NODE(DEPTH_TO_SPACE, luci::CircleDepthToSpace)
CIRCLE_NODE(DEPTHWISE_CONV_2D, luci::CircleDepthwiseConv2D)
CIRCLE_NODE(DIV, luci::CircleDiv)
CIRCLE_NODE(ELU, luci::CircleElu)
CIRCLE_NODE(EQUAL, luci::CircleEqual)
CIRCLE_NODE(EXP, luci::CircleExp)
CIRCLE_NODE(EXPAND_DIMS, luci::CircleExpandDims)
CIRCLE_NODE(FILL, luci::CircleFill)
CIRCLE_NODE(FLOOR, luci::CircleFloor)
CIRCLE_NODE(FLOOR_DIV, luci::CircleFloorDiv)
CIRCLE_NODE(FLOOR_MOD, luci::CircleFloorMod)
CIRCLE_NODE(FULLY_CONNECTED, luci::CircleFullyConnected)
CIRCLE_NODE(GATHER, luci::CircleGather)
CIRCLE_NODE(GATHER_ND, luci::CircleGatherNd)
CIRCLE_NODE(GREATER, luci::CircleGreater)
CIRCLE_NODE(GREATER_EQUAL, luci::CircleGreaterEqual)
CIRCLE_NODE(IF, luci::CircleIf)
CIRCLE_NODE(L2_NORMALIZATION, luci::CircleL2Normalize)
CIRCLE_NODE(L2_POOL_2D, luci::CircleL2Pool2D)
CIRCLE_NODE(LEAKY_RELU, luci::CircleLeakyRelu)
CIRCLE_NODE(LESS, luci::CircleLess)
CIRCLE_NODE(LESS_EQUAL, luci::CircleLessEqual)
CIRCLE_NODE(LOCAL_RESPONSE_NORMALIZATION, luci::CircleLocalResponseNormalization)
CIRCLE_NODE(LOG, luci::CircleLog)
CIRCLE_NODE(LOGICAL_AND, luci::CircleLogicalAnd)
CIRCLE_NODE(LOGICAL_NOT, luci::CircleLogicalNot)
CIRCLE_NODE(LOGICAL_OR, luci::CircleLogicalOr)
CIRCLE_NODE(LOGISTIC, luci::CircleLogistic)
CIRCLE_NODE(LOG_SOFTMAX, luci::CircleLogSoftmax)
CIRCLE_NODE(MATRIX_DIAG, luci::CircleMatrixDiag)
CIRCLE_NODE(MAX_POOL_2D, luci::CircleMaxPool2D)
CIRCLE_NODE(MATRIX_SET_DIAG, luci::CircleMatrixSetDiag)
CIRCLE_NODE(MAXIMUM, luci::CircleMaximum)
CIRCLE_NODE(MEAN, luci::CircleMean)
CIRCLE_NODE(MINIMUM, luci::CircleMinimum)
CIRCLE_NODE(MIRROR_PAD, luci::CircleMirrorPad)
CIRCLE_NODE(MUL, luci::CircleMul)
CIRCLE_NODE(NEG, luci::CircleNeg)
CIRCLE_NODE(NON_MAX_SUPPRESSION_V4, luci::CircleNonMaxSuppressionV4)
CIRCLE_NODE(NON_MAX_SUPPRESSION_V5, luci::CircleNonMaxSuppressionV5)
CIRCLE_NODE(NOT_EQUAL, luci::CircleNotEqual)
CIRCLE_NODE(ONE_HOT, luci::CircleOneHot)
CIRCLE_NODE(PACK, luci::CirclePack)
CIRCLE_NODE(PAD, luci::CirclePad)
CIRCLE_NODE(PADV2, luci::CirclePadV2)
CIRCLE_NODE(POW, luci::CirclePow)
CIRCLE_NODE(PRELU, luci::CirclePRelu)
CIRCLE_NODE(RANGE, luci::CircleRange)
CIRCLE_NODE(RANK, luci::CircleRank)
CIRCLE_NODE(REDUCE_ANY, luci::CircleReduceAny)
CIRCLE_NODE(REDUCE_MAX, luci::CircleReduceMax)
CIRCLE_NODE(REDUCE_MIN, luci::CircleReduceMin)
CIRCLE_NODE(REDUCE_PROD, luci::CircleReduceProd)
CIRCLE_NODE(RELU, luci::CircleRelu)
CIRCLE_NODE(RELU6, luci::CircleRelu6)
CIRCLE_NODE(RELU_N1_TO_1, luci::CircleReluN1To1)
CIRCLE_NODE(RESHAPE, luci::CircleReshape)
CIRCLE_NODE(RESIZE_BILINEAR, luci::CircleResizeBilinear)
CIRCLE_NODE(RESIZE_NEAREST_NEIGHBOR, luci::CircleResizeNearestNeighbor)
CIRCLE_NODE(REVERSE_SEQUENCE, luci::CircleReverseSequence)
CIRCLE_NODE(REVERSE_V2, luci::CircleReverseV2)
CIRCLE_NODE(ROUND, luci::CircleRound)
CIRCLE_NODE(RSQRT, luci::CircleRsqrt)
CIRCLE_NODE(SCATTER_ND, luci::CircleScatterNd)
CIRCLE_NODE(SEGMENT_SUM, luci::CircleSegmentSum)
CIRCLE_NODE(SELECT, luci::CircleSelect)
CIRCLE_NODE(SELECT_V2, luci::CircleSelectV2)
CIRCLE_NODE(SHAPE, luci::CircleShape)
CIRCLE_NODE(SIN, luci::CircleSin)
CIRCLE_NODE(SLICE, luci::CircleSlice)
CIRCLE_NODE(SOFTMAX, luci::CircleSoftmax)
CIRCLE_NODE(SPACE_TO_BATCH_ND, luci::CircleSpaceToBatchND)
CIRCLE_NODE(SPACE_TO_DEPTH, luci::CircleSpaceToDepth)
CIRCLE_NODE(SPARSE_TO_DENSE, luci::CircleSparseToDense)
CIRCLE_NODE(SPLIT, luci::CircleSplit)
CIRCLE_NODE(SPLIT_V, luci::CircleSplitV)
CIRCLE_NODE(SQRT, luci::CircleSqrt)
CIRCLE_NODE(SQUARE, luci::CircleSquare)
CIRCLE_NODE(SQUARED_DIFFERENCE, luci::CircleSquaredDifference)
CIRCLE_NODE(SQUEEZE, luci::CircleSqueeze)
CIRCLE_NODE(STRIDED_SLICE, luci::CircleStridedSlice)
CIRCLE_NODE(SUB, luci::CircleSub)
CIRCLE_NODE(SUM, luci::CircleSum)
CIRCLE_NODE(TANH, luci::CircleTanh)
CIRCLE_NODE(TILE, luci::CircleTile)
CIRCLE_NODE(TOPK_V2, luci::CircleTopKV2)
CIRCLE_NODE(TRANSPOSE, luci::CircleTranspose)
CIRCLE_NODE(TRANSPOSE_CONV, luci::CircleTransposeConv)
CIRCLE_NODE(UNIQUE, luci::CircleUnique)
CIRCLE_NODE(UNPACK, luci::CircleUnpack)
CIRCLE_NODE(WHERE, luci::CircleWhere)
CIRCLE_NODE(WHILE, luci::CircleWhile)
CIRCLE_NODE(ZEROS_LIKE, luci::CircleZerosLike)
// Circle Only
CIRCLE_NODE(BCQ_FULLY_CONNECTED, luci::CircleBCQFullyConnected)
CIRCLE_NODE(BCQ_GATHER, luci::CircleBCQGather)
CIRCLE_NODE(INSTANCE_NORM, luci::CircleInstanceNorm)
// Virtual node(s)
CIRCLE_NODE(CIRCLECONST, luci::CircleConst)
CIRCLE_NODE(CIRCLEINPUT, luci::CircleInput)
CIRCLE_NODE(CIRCLEOUTPUT, luci::CircleOutput)
CIRCLE_NODE(CIRCLEOUTPUTDUMMY, luci::CircleOutputDummy)
CIRCLE_NODE(CIRCLEOUTPUTEXCLUDE, luci::CircleOutputExclude)
CIRCLE_NODE(CIRCLECUSTOMOUT, luci::CircleCustomOut)
CIRCLE_NODE(CIRCLEIFOUT, luci::CircleIfOut)
CIRCLE_NODE(CIRCLENONMAXSUPPRESSIONV4OUT, luci::CircleNonMaxSuppressionV4Out)
CIRCLE_NODE(CIRCLENONMAXSUPPRESSIONV5OUT, luci::CircleNonMaxSuppressionV5Out)
CIRCLE_NODE(CIRCLESPLITOUT, luci::CircleSplitOut)
CIRCLE_NODE(CIRCLESPLITVOUT, luci::CircleSplitVOut)
CIRCLE_NODE(CIRCLETOPKV2OUT, luci::CircleTopKV2Out)
CIRCLE_NODE(CIRCLEUNIQUEOUT, luci::CircleUniqueOut)
CIRCLE_NODE(CIRCLEUNPACKOUT, luci::CircleUnpackOut)
CIRCLE_NODE(CIRCLEWHILEOUT, luci::CircleWhileOut)