compiler/luci/service/include/luci/Service/CircleShapeSignatureInference.h

   1 /*
   2  * Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
   3  *
   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
   7  *
   8  *    http://www.apache.org/licenses/LICENSE-2.0
   9  *
  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.
  15  */
  16
  17 #ifndef __LUCI_CIRCLE_SHAPE_SIGNATURE_INFERENCE_H__
  18 #define __LUCI_CIRCLE_SHAPE_SIGNATURE_INFERENCE_H__
  19
  20 #include <luci/IR/CircleNodes.h>
  21 #include <luci/IR/CircleNodeVisitor.h>
  22 #include <luci/IR/CircleShapeSignature.h>
  23 #include <luci/Service/CircleShapeSignatureInferenceHelper.h>
  24
  25 namespace luci
  26 {
  27
  28 namespace ssinf // namespace for Shape Signature Inference
  29 {
  30
  31 struct Rule
  32 {
  33   bool infer(const luci::CircleNode *, ShapeSignature &) const;
  34 };
  35
  36 class Algorithm final : public luci::CircleNodeVisitor<ShapeSignature>
  37 {
  38 public:
  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(); }
  41
  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;
 152
 153   // Circle Only
 154   // ShapeSignature visit(const luci::CircleBCQFullyConnected *node) final;
 155   // ShapeSignature visit(const luci::CircleBCQGather *node) final;
 156   // ShapeSignature visit(const luci::CircleInstanceNorm *node) final;
 157
 158   // Virtual
 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;
 173 };
 174
 175 } // namespace ssinf
 176
 177 } // namespace luci
 178
 179 #endif // __LUCI_CIRCLE_SHAPE_SIGNATURE_INFERENCE_H__