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
|
#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(ARG_MAX, luci::CircleArgMax)
CIRCLE_NODE(AVERAGE_POOL_2D, luci::CircleAveragePool2D)
CIRCLE_NODE(BATCH_TO_SPACE_ND, luci::CircleBatchToSpaceND)
CIRCLE_NODE(CONCATENATION, luci::CircleConcatenation)
CIRCLE_NODE(CONST, luci::CircleConst)
CIRCLE_NODE(CONV_2D, luci::CircleConv2D)
CIRCLE_NODE(COS, luci::CircleCos)
CIRCLE_NODE(DEPTHWISE_CONV_2D, luci::CircleDepthwiseConv2D)
CIRCLE_NODE(DIV, luci::CircleDiv)
CIRCLE_NODE(EQUAL, luci::CircleEqual)
CIRCLE_NODE(EXP, luci::CircleExp)
CIRCLE_NODE(FULLY_CONNECTED, luci::CircleFullyConnected)
CIRCLE_NODE(GATHER, luci::CircleGather)
CIRCLE_NODE(LOGICAL_NOT, luci::CircleLogicalNot)
CIRCLE_NODE(LOGICAL_OR, luci::CircleLogicalOr)
CIRCLE_NODE(MAXIMUM, luci::CircleMaximum)
CIRCLE_NODE(MAX_POOL_2D, luci::CircleMaxPool2D)
CIRCLE_NODE(MEAN, luci::CircleMean)
CIRCLE_NODE(MUL, luci::CircleMul)
CIRCLE_NODE(PACK, luci::CirclePack)
CIRCLE_NODE(PAD, luci::CirclePad)
CIRCLE_NODE(RELU, luci::CircleRelu)
CIRCLE_NODE(RELU6, luci::CircleRelu6)
CIRCLE_NODE(RESHAPE, luci::CircleReshape)
CIRCLE_NODE(RSQRT, luci::CircleRsqrt)
CIRCLE_NODE(SOFTMAX, luci::CircleSoftmax)
CIRCLE_NODE(SQRT, luci::CircleSqrt)
CIRCLE_NODE(SQUARED_DIFFERENCE, luci::CircleSquaredDifference)
CIRCLE_NODE(SUB, luci::CircleSub)
// TODO TFLTanh
CIRCLE_NODE(TRANSPOSE, luci::CircleTranspose)
CIRCLE_NODE(TRANSPOSE_CONV, luci::CircleTransposeConv)
// Circle Only
CIRCLE_NODE(INSTANCE_NORM, luci::CircleInstanceNorm)
// Virtual node(s)
CIRCLE_NODE(CIRCLEINPUT, luci::CircleInput)
CIRCLE_NODE(CIRCLEOUTPUT, luci::CircleOutput)
|
