| 2383 | } |
| 2384 | |
| 2385 | DirectSession::RunState::RunState( |
| 2386 | const std::vector<string>& pending_input_names, |
| 2387 | const std::vector<string>& pending_output_names, int64 step_id, |
| 2388 | const std::vector<Device*>* devices) |
| 2389 | : step_container(step_id, [devices, step_id](const string& name) { |
| 2390 | for (auto d : *devices) { |
| 2391 | if (!d->resource_manager()->Cleanup(name).ok()) { |
| 2392 | // Do nothing... |
| 2393 | } |
| 2394 | ScopedAllocatorMgr* sam = d->GetScopedAllocatorMgr(); |
| 2395 | if (sam) sam->Cleanup(step_id); |
| 2396 | } |
| 2397 | }) { |
| 2398 | // Initially all the feeds and fetches are pending. |
| 2399 | for (auto& name : pending_input_names) { |
| 2400 | pending_inputs[name] = false; |
| 2401 | } |
| 2402 | for (auto& name : pending_output_names) { |
| 2403 | pending_outputs[name] = false; |
| 2404 | } |
| 2405 | } |
| 2406 | |
| 2407 | DirectSession::RunState::RunState(int64 step_id, |
| 2408 | const std::vector<Device*>* devices) |
nothing calls this directly
no test coverage detected