runtime/onert/core/src/interp/InterpExecutor.h

   1 /*
   2  * Copyright (c) 2019 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 /**
  18  * @file  InterpExecutor.h
  19  * @brief This file contains InterpExecutor class\n
  20  *        to manage interpreter execution and environment
  21  */
  22 #ifndef __ONERT_INTERP_INTERP_EXECUTOR_H__
  23 #define __ONERT_INTERP_INTERP_EXECUTOR_H__
  24
  25 #include "ir/OperandIndexMap.h"
  26 #include "ir/Graph.h"
  27 #include "exec/IExecutor.h"
  28
  29 namespace onert
  30 {
  31 namespace interp
  32 {
  33
  34 class ITensor;
  35
  36 /**
  37  * @brief Class to execute model using interpreter
  38  */
  39 class InterpExecutor final : public exec::IExecutor
  40 {
  41 public:
  42   explicit InterpExecutor(const ir::Graph &graph) : _graph(graph)
  43   {
  44     // DO NOTHING
  45   }
  46
  47 public:
  48   /**
  49    * @brief   Return graph object
  50    * @return  Graph object
  51    */
  52   const ir::Graph &graph() final { return _graph; }
  53   void setIndexedRanks(std::shared_ptr<ir::OperationIndexMap<int64_t>>) override{
  54       // Not implemented
  55   };
  56   /**
  57    * @brief  Start execution
  58    * @note   It should be called after setting input and output buffer
  59    */
  60   void execute(const exec::IODescription &desc) final;
  61
  62 private:
  63   const ir::Graph &_graph;
  64   ir::OperandIndexMap<std::shared_ptr<ITensor>> _tensor_map;
  65 };
  66
  67 } // namespace interp
  68 } // namespace onert
  69
  70 #endif // __ONERT_INTERP_INTERP_EXECUTOR_H__