A merge node for control flow.
| 623 | |
| 624 | // A merge node for control flow. |
| 625 | Node* AddControlMerge(const string& in_name1, const string& in_name2, Graph* g, |
| 626 | const string& node_name, const string& device_name, |
| 627 | Status* status) { |
| 628 | NodeBuilder node_builder(node_name, "Merge", g->op_registry()); |
| 629 | node_builder.Input({{in_name1, 0, DT_FLOAT}, {in_name2, 0, DT_FLOAT}}); |
| 630 | Node* res_node; |
| 631 | *status = node_builder.Finalize(g, &res_node, /*consume=*/true); |
| 632 | if (!status->ok()) return nullptr; |
| 633 | res_node->set_assigned_device_name(device_name); |
| 634 | return res_node; |
| 635 | } |
| 636 | |
| 637 | // A switch node for control flow. |
| 638 | Node* AddControlSwitch(NodeBuilder::NodeOut input1, NodeBuilder::NodeOut input2, |
no test coverage detected