| 251 | } |
| 252 | |
| 253 | Status PrefetchRunnerMgr::StopRunners(std::string graph_key, Session* sess) { |
| 254 | if (prefetch_runners_.count(sess) == 0) |
| 255 | return Status(errors::NotFound("No PrefechRunners run in Session")); |
| 256 | |
| 257 | coords_[sess]->RequestStop(); |
| 258 | for (auto runner : prefetch_runners_[sess]) |
| 259 | runner->Join(); |
| 260 | |
| 261 | coords_.erase(sess); |
| 262 | prefetch_runners_.erase(sess); |
| 263 | |
| 264 | return Status::OK(); |
| 265 | } |
| 266 | |
| 267 | } // end of namespace tensorflow |
no test coverage detected