compiler/luci-interpreter/src/loader/KernelBuilder.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_INTERPRETER_LOADER_KERNELBUILDER_H
  18 #define LUCI_INTERPRETER_LOADER_KERNELBUILDER_H
  19
  20 #include "core/Kernel.h"
  21 #include "core/RuntimeGraph.h"
  22
  23 #include <luci/IR/CircleNodeVisitor.h>
  24
  25 #include <memory>
  26 #include <vector>
  27 #include <unordered_map>
  28
  29 namespace luci_interpreter
  30 {
  31
  32 class KernelBuilder : public luci::CircleNodeVisitor<std::unique_ptr<Kernel>>
  33 {
  34 public:
  35   KernelBuilder(
  36       const std::unordered_map<const loco::Graph *, RuntimeGraph *> &graph_to_runtime_graph,
  37       const std::unordered_map<const loco::Node *, Tensor *> &node_to_tensor)
  38       : _graph_to_runtime_graph(graph_to_runtime_graph), _node_to_tensor(node_to_tensor)
  39   {
  40   }
  41
  42   std::unique_ptr<Kernel> visit(const luci::CircleAdd *node) override;
  43   std::unique_ptr<Kernel> visit(const luci::CircleArgMax *node) override;
  44   std::unique_ptr<Kernel> visit(const luci::CircleAveragePool2D *node) override;
  45   std::unique_ptr<Kernel> visit(const luci::CircleConcatenation *node) override;
  46   std::unique_ptr<Kernel> visit(const luci::CircleConv2D *node) override;
  47   std::unique_ptr<Kernel> visit(const luci::CircleConst *node) override;
  48   std::unique_ptr<Kernel> visit(const luci::CircleDepthToSpace *node) override;
  49   std::unique_ptr<Kernel> visit(const luci::CircleDepthwiseConv2D *node) override;
  50   std::unique_ptr<Kernel> visit(const luci::CircleElu *node) override;
  51   std::unique_ptr<Kernel> visit(const luci::CircleFullyConnected *node) override;
  52   std::unique_ptr<Kernel> visit(const luci::CircleIf *node) override;
  53   std::unique_ptr<Kernel> visit(const luci::CircleL2Normalize *node) override;
  54   std::unique_ptr<Kernel> visit(const luci::CircleL2Pool2D *node) override;
  55   std::unique_ptr<Kernel> visit(const luci::CircleLeakyRelu *node) override;
  56   std::unique_ptr<Kernel> visit(const luci::CircleLocalResponseNormalization *node) override;
  57   std::unique_ptr<Kernel> visit(const luci::CircleLogistic *node) override;
  58   std::unique_ptr<Kernel> visit(const luci::CircleInput *node) override;
  59   std::unique_ptr<Kernel> visit(const luci::CircleMaxPool2D *node) override;
  60   std::unique_ptr<Kernel> visit(const luci::CircleMean *node) override;
  61   std::unique_ptr<Kernel> visit(const luci::CircleMul *node) override;
  62   std::unique_ptr<Kernel> visit(const luci::CircleOutput *node) override;
  63   std::unique_ptr<Kernel> visit(const luci::CirclePad *node) override;
  64   std::unique_ptr<Kernel> visit(const luci::CircleReshape *node) override;
  65   std::unique_ptr<Kernel> visit(const luci::CircleReverseV2 *node) override;
  66   std::unique_ptr<Kernel> visit(const luci::CircleSlice *node) override;
  67   std::unique_ptr<Kernel> visit(const luci::CircleSoftmax *node) override;
  68   std::unique_ptr<Kernel> visit(const luci::CircleSpaceToDepth *node) override;
  69   std::unique_ptr<Kernel> visit(const luci::CircleSplit *node) override;
  70   std::unique_ptr<Kernel> visit(const luci::CircleStridedSlice *node) override;
  71   std::unique_ptr<Kernel> visit(const luci::CircleSqueeze *node) override;
  72   std::unique_ptr<Kernel> visit(const luci::CircleTranspose *node) override;
  73   std::unique_ptr<Kernel> visit(const luci::CircleTransposeConv *node) override;
  74   std::unique_ptr<Kernel> visit(const luci::CircleUnpack *node) override;
  75
  76 private:
  77   const Tensor *getInputTensor(const loco::Node *node) const;
  78
  79   const Tensor *getOptionalInputTensor(const loco::Node *node) const;
  80
  81   Tensor *getOutputTensor(const loco::Node *node) const;
  82
  83   std::vector<Tensor *> getOutputTensors(const std::vector<const loco::Node *> &nodes) const;
  84
  85   RuntimeGraph *getRuntimeGraph(const loco::Graph *graph) const;
  86
  87 private:
  88   const std::unordered_map<const loco::Graph *, RuntimeGraph *> &_graph_to_runtime_graph;
  89   const std::unordered_map<const loco::Node *, Tensor *> &_node_to_tensor;
  90 };
  91
  92 } // namespace luci_interpreter
  93
  94 #endif // LUCI_INTERPRETER_LOADER_KERNELBUILDER_H