| 569 | } |
| 570 | |
| 571 | void Graph::RemoveControlEdge(const Edge* e) { |
| 572 | if (!e->src_->IsSource() && !e->dst_->IsSink()) { |
| 573 | e->dst_->MaybeCopyOnWrite(); |
| 574 | string e_src_name = strings::StrCat("^", e->src_->name()); |
| 575 | auto* inputs = e->dst_->props_->node_def.mutable_input(); |
| 576 | for (auto it = inputs->begin(); it != inputs->end(); ++it) { |
| 577 | if (*it == e_src_name) { |
| 578 | inputs->erase(it); |
| 579 | break; |
| 580 | } |
| 581 | } |
| 582 | } |
| 583 | RemoveEdge(e); |
| 584 | } |
| 585 | |
| 586 | namespace { |
| 587 | const Edge* FindEdge(const Node* dst, int index) { |