diff options
Diffstat (limited to 'compiler/luci/lang/include/luci/IR/CircleNodes.lst')
| -rw-r--r-- | compiler/luci/lang/include/luci/IR/CircleNodes.lst | 266 |
1 files changed, 141 insertions, 125 deletions
diff --git a/compiler/luci/lang/include/luci/IR/CircleNodes.lst b/compiler/luci/lang/include/luci/IR/CircleNodes.lst index 9f0a1b16e..1646909e8 100644 --- a/compiler/luci/lang/include/luci/IR/CircleNodes.lst +++ b/compiler/luci/lang/include/luci/IR/CircleNodes.lst @@ -2,6 +2,10 @@ #error "Define CIRCLE_NODE" #endif // CIRCLE_NODE +#ifndef CIRCLE_VNODE +#error "Define CIRCLE_VNODE" +#endif // CIRCLE_VNODE + // // PLEASE SORT NODE DECLS IN ALPHABETICAL ORDER // @@ -11,130 +15,142 @@ // .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_NODE(ABS, CircleAbs) +CIRCLE_NODE(ADD, CircleAdd) +CIRCLE_NODE(ADD_N, CircleAddN) +CIRCLE_NODE(ARG_MAX, CircleArgMax) +CIRCLE_NODE(ARG_MIN, CircleArgMin) +CIRCLE_NODE(AVERAGE_POOL_2D, CircleAveragePool2D) +CIRCLE_NODE(BATCH_TO_SPACE_ND, CircleBatchToSpaceND) +CIRCLE_NODE(BATCH_MATMUL, CircleBatchMatMul) +CIRCLE_NODE(BIDIRECTIONAL_SEQUENCE_LSTM, CircleBidirectionalSequenceLSTM) +CIRCLE_NODE(CAST, CircleCast) +CIRCLE_NODE(CEIL, CircleCeil) +CIRCLE_NODE(CONCATENATION, CircleConcatenation) +CIRCLE_NODE(CONV_2D, CircleConv2D) +CIRCLE_NODE(COS, CircleCos) +CIRCLE_NODE(CUSTOM, CircleCustom) +CIRCLE_NODE(DENSIFY, CircleDensify) +CIRCLE_NODE(DEPTH_TO_SPACE, CircleDepthToSpace) +CIRCLE_NODE(DEPTHWISE_CONV_2D, CircleDepthwiseConv2D) +CIRCLE_NODE(DEQUANTIZE, CircleDequantize) +CIRCLE_NODE(DIV, CircleDiv) +CIRCLE_NODE(ELU, CircleElu) +CIRCLE_NODE(EQUAL, CircleEqual) +CIRCLE_NODE(EXP, CircleExp) +CIRCLE_NODE(EXPAND_DIMS, CircleExpandDims) +CIRCLE_NODE(FAKE_QUANT, CircleFakeQuant) +CIRCLE_NODE(FILL, CircleFill) +CIRCLE_NODE(FLOOR, CircleFloor) +CIRCLE_NODE(FLOOR_DIV, CircleFloorDiv) +CIRCLE_NODE(FLOOR_MOD, CircleFloorMod) +CIRCLE_NODE(FULLY_CONNECTED, CircleFullyConnected) +CIRCLE_NODE(GATHER, CircleGather) +CIRCLE_NODE(GATHER_ND, CircleGatherNd) +CIRCLE_NODE(GELU, CircleGelu) +CIRCLE_NODE(GREATER, CircleGreater) +CIRCLE_NODE(GREATER_EQUAL, CircleGreaterEqual) +CIRCLE_NODE(HARD_SWISH, CircleHardSwish) +CIRCLE_NODE(IF, CircleIf) +CIRCLE_NODE(L2_NORMALIZATION, CircleL2Normalize) +CIRCLE_NODE(L2_POOL_2D, CircleL2Pool2D) +CIRCLE_NODE(LEAKY_RELU, CircleLeakyRelu) +CIRCLE_NODE(LESS, CircleLess) +CIRCLE_NODE(LESS_EQUAL, CircleLessEqual) +CIRCLE_NODE(LOCAL_RESPONSE_NORMALIZATION, CircleLocalResponseNormalization) +CIRCLE_NODE(LOG, CircleLog) +CIRCLE_NODE(LOGICAL_AND, CircleLogicalAnd) +CIRCLE_NODE(LOGICAL_NOT, CircleLogicalNot) +CIRCLE_NODE(LOGICAL_OR, CircleLogicalOr) +CIRCLE_NODE(LOGISTIC, CircleLogistic) +CIRCLE_NODE(LOG_SOFTMAX, CircleLogSoftmax) +CIRCLE_NODE(MATRIX_DIAG, CircleMatrixDiag) +CIRCLE_NODE(MAX_POOL_2D, CircleMaxPool2D) +CIRCLE_NODE(MATRIX_SET_DIAG, CircleMatrixSetDiag) +CIRCLE_NODE(MAXIMUM, CircleMaximum) +CIRCLE_NODE(MEAN, CircleMean) +CIRCLE_NODE(MINIMUM, CircleMinimum) +CIRCLE_NODE(MIRROR_PAD, CircleMirrorPad) +CIRCLE_NODE(MUL, CircleMul) +CIRCLE_NODE(NEG, CircleNeg) +CIRCLE_NODE(NON_MAX_SUPPRESSION_V4, CircleNonMaxSuppressionV4) +CIRCLE_NODE(NON_MAX_SUPPRESSION_V5, CircleNonMaxSuppressionV5) +CIRCLE_NODE(NOT_EQUAL, CircleNotEqual) +CIRCLE_NODE(ONE_HOT, CircleOneHot) +CIRCLE_NODE(PACK, CirclePack) +CIRCLE_NODE(PAD, CirclePad) +CIRCLE_NODE(PADV2, CirclePadV2) +CIRCLE_NODE(POW, CirclePow) +CIRCLE_NODE(PRELU, CirclePRelu) +CIRCLE_NODE(QUANTIZE, CircleQuantize) +CIRCLE_NODE(RANGE, CircleRange) +CIRCLE_NODE(RANK, CircleRank) +CIRCLE_NODE(REDUCE_ANY, CircleReduceAny) +CIRCLE_NODE(REDUCE_MAX, CircleReduceMax) +CIRCLE_NODE(REDUCE_MIN, CircleReduceMin) +CIRCLE_NODE(REDUCE_PROD, CircleReduceProd) +CIRCLE_NODE(RELU, CircleRelu) +CIRCLE_NODE(RELU6, CircleRelu6) +CIRCLE_NODE(RELU_N1_TO_1, CircleReluN1To1) +CIRCLE_NODE(RESHAPE, CircleReshape) +CIRCLE_NODE(RESIZE_BILINEAR, CircleResizeBilinear) +CIRCLE_NODE(RESIZE_NEAREST_NEIGHBOR, CircleResizeNearestNeighbor) +CIRCLE_NODE(REVERSE_SEQUENCE, CircleReverseSequence) +CIRCLE_NODE(REVERSE_V2, CircleReverseV2) +CIRCLE_NODE(ROUND, CircleRound) +CIRCLE_NODE(RSQRT, CircleRsqrt) +CIRCLE_NODE(SCATTER_ND, CircleScatterNd) +CIRCLE_NODE(SEGMENT_SUM, CircleSegmentSum) +CIRCLE_NODE(SELECT, CircleSelect) +CIRCLE_NODE(SELECT_V2, CircleSelectV2) +CIRCLE_NODE(SHAPE, CircleShape) +CIRCLE_NODE(SIN, CircleSin) +CIRCLE_NODE(SLICE, CircleSlice) +CIRCLE_NODE(SOFTMAX, CircleSoftmax) +CIRCLE_NODE(SPACE_TO_BATCH_ND, CircleSpaceToBatchND) +CIRCLE_NODE(SPACE_TO_DEPTH, CircleSpaceToDepth) +CIRCLE_NODE(SPARSE_TO_DENSE, CircleSparseToDense) +CIRCLE_NODE(SPLIT, CircleSplit) +CIRCLE_NODE(SPLIT_V, CircleSplitV) +CIRCLE_NODE(SQRT, CircleSqrt) +CIRCLE_NODE(SQUARE, CircleSquare) +CIRCLE_NODE(SQUARED_DIFFERENCE, CircleSquaredDifference) +CIRCLE_NODE(SQUEEZE, CircleSqueeze) +CIRCLE_NODE(STRIDED_SLICE, CircleStridedSlice) +CIRCLE_NODE(SUB, CircleSub) +CIRCLE_NODE(SUM, CircleSum) +CIRCLE_NODE(SVDF, CircleSVDF) +CIRCLE_NODE(TANH, CircleTanh) +CIRCLE_NODE(TILE, CircleTile) +CIRCLE_NODE(TOPK_V2, CircleTopKV2) +CIRCLE_NODE(TRANSPOSE, CircleTranspose) +CIRCLE_NODE(TRANSPOSE_CONV, CircleTransposeConv) +CIRCLE_NODE(UNIDIRECTIONAL_SEQUENCE_LSTM, CircleUnidirectionalSequenceLSTM) +CIRCLE_NODE(UNIQUE, CircleUnique) +CIRCLE_NODE(UNPACK, CircleUnpack) +CIRCLE_NODE(WHERE, CircleWhere) +CIRCLE_NODE(WHILE, CircleWhile) +CIRCLE_NODE(ZEROS_LIKE, CircleZerosLike) // Circle Only -CIRCLE_NODE(BCQ_FULLY_CONNECTED, luci::CircleBCQFullyConnected) -CIRCLE_NODE(BCQ_GATHER, luci::CircleBCQGather) -CIRCLE_NODE(INSTANCE_NORM, luci::CircleInstanceNorm) +CIRCLE_NODE(BCQ_FULLY_CONNECTED, CircleBCQFullyConnected) +CIRCLE_NODE(BCQ_GATHER, CircleBCQGather) +CIRCLE_NODE(INSTANCE_NORM, 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) +CIRCLE_VNODE(CIRCLECONST, CircleConst) +CIRCLE_VNODE(CIRCLEINPUT, CircleInput) +CIRCLE_VNODE(CIRCLEOUTPUT, CircleOutput) +CIRCLE_VNODE(CIRCLEOUTPUTDUMMY, CircleOutputDummy) +CIRCLE_VNODE(CIRCLEOUTPUTEXCLUDE, CircleOutputExclude) +CIRCLE_VNODE(CIRCLEVARIABLE, CircleVariable) +// Multi-output virtual nodes +CIRCLE_VNODE(CIRCLEBIDIRECTIONAL_SEQUENCE_LSTM_OUT, CircleBidirectionalSequenceLSTMOut) +CIRCLE_VNODE(CIRCLECUSTOMOUT, CircleCustomOut) +CIRCLE_VNODE(CIRCLEIFOUT, CircleIfOut) +CIRCLE_VNODE(CIRCLENONMAXSUPPRESSIONV4OUT, CircleNonMaxSuppressionV4Out) +CIRCLE_VNODE(CIRCLENONMAXSUPPRESSIONV5OUT, CircleNonMaxSuppressionV5Out) +CIRCLE_VNODE(CIRCLESPLITOUT, CircleSplitOut) +CIRCLE_VNODE(CIRCLESPLITVOUT, CircleSplitVOut) +CIRCLE_VNODE(CIRCLETOPKV2OUT, CircleTopKV2Out) +CIRCLE_VNODE(CIRCLEUNIQUEOUT, CircleUniqueOut) +CIRCLE_VNODE(CIRCLEUNPACKOUT, CircleUnpackOut) +CIRCLE_VNODE(CIRCLEWHILEOUT, CircleWhileOut) |