| 1294 | } |
| 1295 | |
| 1296 | Status GraphConstructor::UpdateVersionDef() { |
| 1297 | if (versions() == nullptr) return Status::OK(); |
| 1298 | |
| 1299 | if (!opts_.importing) { |
| 1300 | g_->set_versions(*versions()); |
| 1301 | return Status::OK(); |
| 1302 | } |
| 1303 | VersionDef g_versions = g_->versions(); |
| 1304 | g_versions.set_producer( |
| 1305 | std::min(g_versions.producer(), versions()->producer())); |
| 1306 | g_versions.set_min_consumer( |
| 1307 | std::max(g_versions.min_consumer(), versions()->min_consumer())); |
| 1308 | if (versions()->bad_consumers_size() > 0) { |
| 1309 | std::set<int> bad(g_versions.bad_consumers().begin(), |
| 1310 | g_versions.bad_consumers().end()); |
| 1311 | bad.insert(versions()->bad_consumers().begin(), |
| 1312 | versions()->bad_consumers().end()); |
| 1313 | g_versions.clear_bad_consumers(); |
| 1314 | for (int v : bad) { |
| 1315 | g_versions.add_bad_consumers(v); |
| 1316 | } |
| 1317 | } |
| 1318 | g_->set_versions(g_versions); |
| 1319 | return Status::OK(); |
| 1320 | } |
| 1321 | |
| 1322 | Status GraphConstructor::PopulateReturnTensors() { |
| 1323 | if (opts_.return_tensors.empty()) return Status::OK(); |