2 * Copyright (c) 2023 Samsung Electronics Co., Ltd. All Rights Reserved
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
8 * http://www.apache.org/licenses/LICENSE-2.0
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.
17 #ifndef __ONERT_COMPILER_TRAIN_TRAINABLE_CODE_MAP_H__
18 #define __ONERT_COMPILER_TRAIN_TRAINABLE_CODE_MAP_H__
20 #include <unordered_map>
21 #include "compiler/OperationLowerInfo.h"
22 #include "exec/train/TrainableFnSequence.h"
23 #include "ir/train/ITrainableOperation.h"
32 struct TrainableCodeAndInfo
34 ir::OperationIndex op_ind;
35 const ir::train::ITrainableOperation *op;
36 const OperationLowerInfo *lower_info;
37 // TODO Change to TrainableFnSequence
38 std::unique_ptr<exec::train::TrainableFnSequence> tn_seq;
40 TrainableCodeAndInfo(const ir::OperationIndex op_ind, const ir::train::ITrainableOperation *op,
41 const OperationLowerInfo *lower_info,
42 std::unique_ptr<exec::train::TrainableFnSequence> &&tn_seq)
43 : op_ind{op_ind}, op{op}, lower_info{lower_info}, tn_seq{std::move(tn_seq)}
48 using TrainableCodeMap = std::unordered_map<ir::OperationIndex, TrainableCodeAndInfo>;
51 } // namespace compiler
54 #endif // __ONERT_COMPILER_TRAIN_TRAINABLE_CODE_MAP_H__