Searched refs:recv_at_host_node (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 126 Node* recv_at_host_node = g->AddNode(recv_at_host_def, &s); in BuildRecvAtHostNode() local
128 return recv_at_host_node; in BuildRecvAtHostNode()
140 Node * recv_at_host_node, in ReplaceArgNodesWithRecvAtHostNode()
156 g->AddControlEdge(recv_at_host_node, edge.dst); in ReplaceArgNodesWithRecvAtHostNode()
158 g->AddEdge(recv_at_host_node, index, edge.dst, edge.dst_input); in ReplaceArgNodesWithRecvAtHostNode()
172 absl::StrCat(recv_at_host_node->name(), ":", index); in ReplaceArgNodesWithRecvAtHostNode()
184 g->AddEdge(key_placeholder, 0, recv_at_host_node, 0); in ReplaceArgNodesWithRecvAtHostNode()
185 return recv_at_host_node; in ReplaceArgNodesWithRecvAtHostNode()
2224 Node * recv_at_host_node, in operator ()()
2262 (*graph)->AddControlEdge(recv_at_host_node, n); in operator ()()
|