| 1588 | |
| 1589 | typedef std::vector<Node*> node_list; |
| 1590 | node_list DuplicateNode(Node* n, Graph* dest, int32 duplicate_num) { |
| 1591 | node_list duplicated_nodes; |
| 1592 | duplicated_nodes.reserve(duplicate_num); |
| 1593 | for (int i = 0; i < duplicate_num; ++i) { |
| 1594 | NodeDef node_def = n->def(); |
| 1595 | node_def.set_name(node_def.name() + "/dup" + std::to_string(i)); |
| 1596 | Status s; |
| 1597 | Node* dup = dest->AddNode(node_def, &s); |
| 1598 | TF_CHECK_OK(s); |
| 1599 | dup->set_assigned_device_name(n->assigned_device_name()); |
| 1600 | duplicated_nodes.emplace_back(dup); |
| 1601 | } |
| 1602 | return duplicated_nodes; |
| 1603 | } |
| 1604 | |
| 1605 | struct EdgeWrapper { |
| 1606 | Node* src; |
no test coverage detected