@@ -89,10 +89,10 @@ static bool VarCanBeDeleted(const std::string &name, const BlockDesc &block,
89
89
type == proto::VarType::LOD_TENSOR_ARRAY;
90
90
}
91
91
92
- std::unordered_map<OperatorBase *, std::vector<std::string>> GetUnusedVars (
93
- const BlockDesc &block,
94
- const std::vector<std::unique_ptr<OperatorBase>> &ops,
95
- const std::vector<std::string> &skip_var_list) {
92
+ std::unordered_map<const OperatorBase *, std::vector<std::string>>
93
+ GetUnusedVars ( const BlockDesc &block,
94
+ const std::vector<std::unique_ptr<OperatorBase>> &ops,
95
+ const std::vector<std::string> &skip_var_list) {
96
96
std::unordered_set<std::string> skip_vars (skip_var_list.begin (),
97
97
skip_var_list.end ());
98
98
@@ -134,7 +134,7 @@ std::unordered_map<OperatorBase *, std::vector<std::string>> GetUnusedVars(
134
134
}
135
135
}
136
136
137
- std::unordered_map<OperatorBase *, std::vector<std::string>> result;
137
+ std::unordered_map<const OperatorBase *, std::vector<std::string>> result;
138
138
for (auto &name_op_idx_pair : var_op_idx_map) {
139
139
auto &name = name_op_idx_pair.first ;
140
140
size_t op_idx = name_op_idx_pair.second ;
@@ -144,8 +144,8 @@ std::unordered_map<OperatorBase *, std::vector<std::string>> GetUnusedVars(
144
144
}
145
145
146
146
void DeleteUnusedTensors (
147
- const Scope &scope, OperatorBase *op,
148
- const std::unordered_map<OperatorBase *, std::vector<std::string>>
147
+ const Scope &scope, const OperatorBase *op,
148
+ const std::unordered_map<const OperatorBase *, std::vector<std::string>>
149
149
&delete_vars_map,
150
150
GarbageCollector *gc) {
151
151
auto iter = delete_vars_map.find (op);
0 commit comments