| 392 | } |
| 393 | |
| 394 | Status LowerWhileHelper::UpdateConsumers() { |
| 395 | for (const Edge* e : while_op_->out_edges()) { |
| 396 | if (e->IsControlEdge()) { |
| 397 | graph_->AddControlEdge(lowered_while_executed_, e->dst()); |
| 398 | } else { |
| 399 | // Feed the outputs directly from the exit nodes so that downstream ops |
| 400 | // can start before all the outputs have been computed. |
| 401 | graph_->AddEdge(exit_nodes_[e->src_output()], 0, e->dst(), |
| 402 | e->dst_input()); |
| 403 | } |
| 404 | } |
| 405 | return Status::OK(); |
| 406 | } |
| 407 | |
| 408 | string LowerWhileHelper::NewName(const string& infix) { |
| 409 | return graph_->NewName(strings::StrCat(name_, "/", infix)); |
nothing calls this directly
no test coverage detected