1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
/*
* Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef __ONERT_BACKEND_BACKEND_CONTEXT_H__
#define __ONERT_BACKEND_BACKEND_CONTEXT_H__
#include <memory>
#include "ir/Graph.h"
namespace onert
{
namespace backend
{
class Backend;
class IConstantInitializer;
class IKernelGenerator;
class IShapeFixer;
class ITensorRegister;
struct ITensorBuilder;
struct IOptimizer;
class BackendContext
{
public:
struct OperationInfo
{
ir::OperationIndex index;
ir::Layout layout;
OperationInfo(ir::OperationIndex index, ir::Layout layout) : index{index}, layout{layout} {}
};
public:
BackendContext(const Backend *backend, const ir::Graph *graph,
std::shared_ptr<ITensorBuilder> tensor_builder = nullptr,
std::shared_ptr<IConstantInitializer> constant_initializer = nullptr,
std::shared_ptr<IKernelGenerator> kernel_gen = nullptr,
std::shared_ptr<IShapeFixer> shape_fixer = nullptr,
std::shared_ptr<ITensorRegister> tensor_register = nullptr,
std::shared_ptr<IOptimizer> optimizer = nullptr)
: _backend{backend}, _graph{graph}, tensor_builder{tensor_builder},
constant_initializer{constant_initializer}, kernel_gen{kernel_gen},
shape_fixer{shape_fixer}, tensor_register{tensor_register}, optimizer{optimizer}
{
}
void initialize(const std::vector<OperationInfo> &operation_list,
const std::vector<ir::OperandIndex> &operand_list);
void fixShapes();
void initConsts();
const Backend *backend() const { return _backend; }
const ir::Graph *graph() const { return _graph; }
const std::vector<OperationInfo> &operation_list() { return _operation_list; }
const std::vector<ir::OperandIndex> &operand_list() { return _operand_list; }
private:
const Backend *_backend{nullptr};
const ir::Graph *_graph{nullptr};
std::vector<OperationInfo> _operation_list;
std::vector<ir::OperandIndex> _operand_list;
public:
std::shared_ptr<ITensorBuilder> tensor_builder;
std::shared_ptr<IConstantInitializer> constant_initializer;
std::shared_ptr<IKernelGenerator> kernel_gen;
std::shared_ptr<IShapeFixer> shape_fixer;
std::shared_ptr<ITensorRegister> tensor_register;
std::shared_ptr<IOptimizer> optimizer;
};
using BackendContexts = std::unordered_map<const Backend *, std::unique_ptr<BackendContext>>;
} // namespace backend
} // namespace onert
#endif // __ONERT_BACKEND_BACKEND_CONTEXT_H__
|
