runtime/onert/backend/acl_cl/BackendContext.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 __ONERT_BACKEND_ACL_CL_BACKEND_CONTEXT_H__
  18 #define __ONERT_BACKEND_ACL_CL_BACKEND_CONTEXT_H__
  19
  20 #include <backend/BackendContext.h>
  21 #include "TensorBuilder.h"
  22 #include "ConstantInitializer.h"
  23 #include "KernelGenerator.h"
  24
  25 namespace onert
  26 {
  27 namespace backend
  28 {
  29 namespace acl_cl
  30 {
  31
  32 class Optimizer;
  33
  34 class BackendContext : public onert::backend::BackendContext
  35 {
  36 public:
  37   BackendContext(const Backend *backend, const ir::Graph *graph,
  38                  std::shared_ptr<ITensorRegistry> tensor_registry = nullptr,
  39                  std::shared_ptr<TensorBuilder> tensor_builder = nullptr,
  40                  std::shared_ptr<ConstantInitializer> constant_initializer = nullptr,
  41                  std::shared_ptr<KernelGenerator> kernel_gen = nullptr)
  42       : onert::backend::BackendContext(backend, graph, tensor_registry),
  43         tensor_builder{tensor_builder}, constant_initializer{constant_initializer},
  44         kernel_gen{kernel_gen}
  45   {
  46   }
  47
  48   ITensorRegistry *genTensors(const std::vector<onert::ir::OpSequenceIndex> &order,
  49                               const ir::OpSequences &op_seqs,
  50                               const ir::LowerInfoMap &lower_info) override;
  51   FunctionMap genKernels(const std::vector<onert::ir::OpSequenceIndex> &order,
  52                          const ir::OpSequences &op_seqs) override;
  53
  54 private:
  55   void initConsts();
  56   void planTensors(const std::vector<onert::ir::OpSequenceIndex> &order,
  57                    const ir::OpSequences &op_seqs, const ir::LowerInfoMap &lower_info);
  58
  59 public:
  60   std::shared_ptr<TensorBuilder> tensor_builder;
  61   std::shared_ptr<ConstantInitializer> constant_initializer;
  62   std::shared_ptr<KernelGenerator> kernel_gen;
  63   std::shared_ptr<Optimizer> optimizer;
  64 };
  65
  66 } // namespace acl_cl
  67 } // namespace backend
  68 } // namespace onert
  69
  70 #endif // __ONERT_BACKEND_ACL_CL_BACKEND_CONTEXT_H__