compiler/luci-interpreter/src/core/RuntimeGraph.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_CORE_RUNTIMEGRAPH_H
  18 #define LUCI_INTERPRETER_CORE_RUNTIMEGRAPH_H
  19
  20 #include "luci_interpreter/core/Tensor.h"
  21 #include "core/Kernel.h"
  22
  23 #include <memory>
  24 #include <vector>
  25
  26 namespace luci_interpreter
  27 {
  28
  29 class RuntimeModule;
  30
  31 class RuntimeGraph
  32 {
  33 public:
  34   explicit RuntimeGraph(RuntimeModule *owning_module) : _owning_module(owning_module) {}
  35
  36   Tensor *addTensor(std::unique_ptr<Tensor> &&tensor);
  37
  38   void setInputTensors(const std::vector<Tensor *> &input_tensors);
  39   void setOutputTensors(const std::vector<Tensor *> &output_tensors);
  40
  41   const std::vector<Tensor *> &getInputTensors() const { return _input_tensors; }
  42   const std::vector<Tensor *> &getOutputTensors() const { return _output_tensors; }
  43
  44   void addKernel(std::unique_ptr<Kernel> &&kernel);
  45
  46   void execute() const;
  47
  48 private:
  49   RuntimeModule *_owning_module;
  50   std::vector<std::unique_ptr<Tensor>> _tensors;
  51   std::vector<Tensor *> _input_tensors;
  52   std::vector<Tensor *> _output_tensors;
  53
  54   // Kernels in execution order.
  55   std::vector<std::unique_ptr<Kernel>> _kernels;
  56 };
  57
  58 } // namespace luci_interpreter
  59
  60 #endif // LUCI_INTERPRETER_CORE_RUNTIMEGRAPH_H