2 * Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
17 #ifndef __LUCI_CIRCLE_SHAPE_SIGNATURE_INFERENCE_H__
18 #define __LUCI_CIRCLE_SHAPE_SIGNATURE_INFERENCE_H__
20 #include <luci/IR/CircleNodes.h>
21 #include <luci/IR/CircleNodeVisitor.h>
22 #include <luci/IR/CircleShapeSignature.h>
23 #include <luci/Service/CircleShapeSignatureInferenceHelper.h>
28 namespace ssinf // namespace for Shape Signature Inference
33 bool infer(const luci::CircleNode *, ShapeSignature &) const;
36 class Algorithm final : public luci::CircleNodeVisitor<ShapeSignature>
39 // TODO Remove this when visit function is implemented for all the operations.
40 ShapeSignature visit(const luci::CircleNode *node) final { return node->shape_signature(); }
42 // ShapeSignature visit(const luci::CircleAbs *node) final;
43 // ShapeSignature visit(const luci::CircleAdd *node) final;
44 // ShapeSignature visit(const luci::CircleAddN *node) final;
45 // ShapeSignature visit(const luci::CircleArgMax *node) final;
46 // ShapeSignature visit(const luci::CircleArgMin *node) final;
47 // ShapeSignature visit(const luci::CircleAveragePool2D *node) final;
48 // ShapeSignature visit(const luci::CircleBatchMatMul *node) final;
49 // ShapeSignature visit(const luci::CircleBatchToSpaceND *node) final;
50 // ShapeSignature visit(const luci::CircleCast *node) final;
51 // ShapeSignature visit(const luci::CircleCeil *node) final;
52 // ShapeSignature visit(const luci::CircleConcatenation *node) final;
53 // ShapeSignature visit(const luci::CircleConst *node) final;
54 // ShapeSignature visit(const luci::CircleConv2D *node) final;
55 // ShapeSignature visit(const luci::CircleCos *node) final;
56 // ShapeSignature visit(const luci::CircleCustom *node) final;
57 // ShapeSignature visit(const luci::CircleDepthToSpace *node) final;
58 // ShapeSignature visit(const luci::CircleDepthwiseConv2D *node) final;
59 // ShapeSignature visit(const luci::CircleDequantize *node) final;
60 // ShapeSignature visit(const luci::CircleDiv *node) final;
61 // ShapeSignature visit(const luci::CircleElu *node) final;
62 // ShapeSignature visit(const luci::CircleEqual *node) final;
63 // ShapeSignature visit(const luci::CircleExp *node) final;
64 // ShapeSignature visit(const luci::CircleExpandDims *node) final;
65 // ShapeSignature visit(const luci::CircleFill *node) final;
66 // ShapeSignature visit(const luci::CircleFloor *node) final;
67 // ShapeSignature visit(const luci::CircleFloorDiv *node) final;
68 // ShapeSignature visit(const luci::CircleFloorMod *node) final;
69 // ShapeSignature visit(const luci::CircleFullyConnected *node) final;
70 // ShapeSignature visit(const luci::CircleGather *node) final;
71 // ShapeSignature visit(const luci::CircleGatherNd *node) final;
72 // ShapeSignature visit(const luci::CircleGreater *node) final;
73 // ShapeSignature visit(const luci::CircleGreaterEqual *node) final;
74 // ShapeSignature visit(const luci::CircleIf *node) final;
75 // ShapeSignature visit(const luci::CircleL2Normalize *node) final;
76 // ShapeSignature visit(const luci::CircleL2Pool2D *node) final;
77 // ShapeSignature visit(const luci::CircleLeakyRelu *node) final;
78 // ShapeSignature visit(const luci::CircleLess *node) final;
79 // ShapeSignature visit(const luci::CircleLessEqual *node) final;
80 // ShapeSignature visit(const luci::CircleLocalResponseNormalization *node) final;
81 // ShapeSignature visit(const luci::CircleLog *node) final;
82 // ShapeSignature visit(const luci::CircleLogicalAnd *node) final;
83 // ShapeSignature visit(const luci::CircleLogicalNot *node) final;
84 // ShapeSignature visit(const luci::CircleLogicalOr *node) final;
85 // ShapeSignature visit(const luci::CircleLogistic *node) final;
86 // ShapeSignature visit(const luci::CircleLogSoftmax *node) final;
87 // ShapeSignature visit(const luci::CircleMatrixDiag *node) final;
88 // ShapeSignature visit(const luci::CircleMatrixSetDiag *node) final;
89 // ShapeSignature visit(const luci::CircleMaximum *node) final;
90 // ShapeSignature visit(const luci::CircleMaxPool2D *node) final;
91 ShapeSignature visit(const luci::CircleMean *node) final;
92 // ShapeSignature visit(const luci::CircleMinimum *node) final;
93 // ShapeSignature visit(const luci::CircleMirrorPad *node) final;
94 // ShapeSignature visit(const luci::CircleNeg *node) final;
95 // ShapeSignature visit(const luci::CircleNonMaxSuppressionV4 *node) final;
96 // ShapeSignature visit(const luci::CircleNonMaxSuppressionV5 *node) final;
97 // ShapeSignature visit(const luci::CircleNotEqual *node) final;
98 // ShapeSignature visit(const luci::CirclePack *node) final;
99 // ShapeSignature visit(const luci::CirclePad *node) final;
100 // ShapeSignature visit(const luci::CirclePadV2 *node) final;
101 // ShapeSignature visit(const luci::CirclePow *node) final;
102 // ShapeSignature visit(const luci::CirclePRelu *node) final;
103 // ShapeSignature visit(const luci::CircleRange *node) final;
104 // ShapeSignature visit(const luci::CircleRank *node) final;
105 // ShapeSignature visit(const luci::CircleMul *node) final;
106 // ShapeSignature visit(const luci::CircleOneHot *node) final;
107 ShapeSignature visit(const luci::CircleReduceAny *node) final;
108 ShapeSignature visit(const luci::CircleReduceMax *node) final;
109 ShapeSignature visit(const luci::CircleReduceMin *node) final;
110 ShapeSignature visit(const luci::CircleReduceProd *node) final;
111 ShapeSignature visit(const luci::CircleRelu *node) final;
112 ShapeSignature visit(const luci::CircleRelu6 *node) final;
113 ShapeSignature visit(const luci::CircleReluN1To1 *node) final;
114 // ShapeSignature visit(const luci::CircleReshape *node) final;
115 // ShapeSignature visit(const luci::CircleResizeBilinear *node) final;
116 // ShapeSignature visit(const luci::CircleResizeNearestNeighbor *node) final;
117 // ShapeSignature visit(const luci::CircleReverseSequence *node) final;
118 // ShapeSignature visit(const luci::CircleReverseV2 *node) final;
119 // ShapeSignature visit(const luci::CircleRound *node) final;
120 // ShapeSignature visit(const luci::CircleRsqrt *node) final;
121 // ShapeSignature visit(const luci::CircleScatterNd *node) final;
122 // ShapeSignature visit(const luci::CircleSegmentSum *node) final;
123 // ShapeSignature visit(const luci::CircleSelect *node) final;
124 // ShapeSignature visit(const luci::CircleSelectV2 *node) final;
125 // ShapeSignature visit(const luci::CircleShape *node) final;
126 // ShapeSignature visit(const luci::CircleSin *node) final;
127 // ShapeSignature visit(const luci::CircleSlice *node) final;
128 // ShapeSignature visit(const luci::CircleSoftmax *node) final;
129 // ShapeSignature visit(const luci::CircleSpaceToBatchND *node) final;
130 // ShapeSignature visit(const luci::CircleSpaceToDepth *node) final;
131 // ShapeSignature visit(const luci::CircleSparseToDense *node) final;
132 // ShapeSignature visit(const luci::CircleSplit *node) final;
133 // ShapeSignature visit(const luci::CircleSplitV *node) final;
134 // ShapeSignature visit(const luci::CircleSqrt *node) final;
135 // ShapeSignature visit(const luci::CircleSquare *node) final;
136 // ShapeSignature visit(const luci::CircleSquaredDifference *node) final;
137 // ShapeSignature visit(const luci::CircleSqueeze *node) final;
138 // ShapeSignature visit(const luci::CircleStridedSlice *node) final;
139 // ShapeSignature visit(const luci::CircleSub *node) final;
140 ShapeSignature visit(const luci::CircleSum *node) final;
141 // ShapeSignature visit(const luci::CircleTanh *node) final;
142 // ShapeSignature visit(const luci::CircleTile *node) final;
143 // ShapeSignature visit(const luci::CircleTopKV2 *node) final;
144 // ShapeSignature visit(const luci::CircleTranspose *node) final;
145 // ShapeSignature visit(const luci::CircleTransposeConv *node) final;
146 // ShapeSignature visit(const luci::CircleUnidirectionalSequenceLSTM *node) final;
147 // ShapeSignature visit(const luci::CircleUnique *node) final;
148 // ShapeSignature visit(const luci::CircleUnpack *node) final;
149 // ShapeSignature visit(const luci::CircleWhere *node) final ;
150 // ShapeSignature visit(const luci::CircleWhile *node) final;
151 // ShapeSignature visit(const luci::CircleZerosLike *node) final;
154 // ShapeSignature visit(const luci::CircleBCQFullyConnected *node) final;
155 // ShapeSignature visit(const luci::CircleBCQGather *node) final;
156 // ShapeSignature visit(const luci::CircleInstanceNorm *node) final;
159 ShapeSignature visit(const luci::CircleInput *node) final;
160 ShapeSignature visit(const luci::CircleOutput *node) final;
161 ShapeSignature visit(const luci::CircleOutputDummy *node) final;
162 ShapeSignature visit(const luci::CircleOutputExclude *node) final;
163 // ShapeSignature visit(const luci::CircleCustomOut *node) final;
164 // ShapeSignature visit(const luci::CircleIfOut *node) final;
165 // ShapeSignature visit(const luci::CircleNonMaxSuppressionV4Out *node) final;
166 // ShapeSignature visit(const luci::CircleNonMaxSuppressionV5Out *node) final;
167 // ShapeSignature visit(const luci::CircleSplitOut *node) final;
168 // ShapeSignature visit(const luci::CircleSplitVOut *node) final;
169 // ShapeSignature visit(const luci::CircleTopKV2Out *node) final;
170 // ShapeSignature visit(const luci::CircleUniqueOut *node) final;
171 // ShapeSignature visit(const luci::CircleUnpackOut *node) final;
172 // ShapeSignature visit(const luci::CircleWhileOut *node) final;
179 #endif // __LUCI_CIRCLE_SHAPE_SIGNATURE_INFERENCE_H__