| 925 | } |
| 926 | |
| 927 | Status FunctionalizeCond::AddIdentityNode(const Node* replacee, Node* if_node, |
| 928 | int port) { |
| 929 | NodeBuilder id_builder(replacee->name(), "Identity"); |
| 930 | id_builder.Input(if_node, port); |
| 931 | string outside_compilation; |
| 932 | if (GetNodeAttr(if_node->def(), kXlaOutsideCompilationAttrName, |
| 933 | &outside_compilation) |
| 934 | .ok()) { |
| 935 | id_builder.Attr(kXlaOutsideCompilationAttrName, outside_compilation); |
| 936 | } |
| 937 | Node* id; |
| 938 | TF_RETURN_IF_ERROR(id_builder.Finalize(graph_, &id)); |
| 939 | state_map_.ResetCondId(id, state_map_.LookupCondId(if_node)); |
| 940 | state_map_.ResetAncestorId(id, state_map_.LookupAncestorId(if_node)); |
| 941 | return Status::OK(); |
| 942 | } |
| 943 | |
| 944 | StatusOr<Node*> FunctionalizeCond::AddIfNode(const NodeDef& def, |
| 945 | const Node* replacee, |
no test coverage detected