Searched refs:send_from_host_node (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 243 Node* send_from_host_node = g->AddNode(send_from_host_def, &s); in BuildSendFromHostNode() local
245 return send_from_host_node; in BuildSendFromHostNode()
257 Node * send_from_host_node, in ReplaceRetNodesWithSendFromHostNode()
265 g->AddControlEdge(edge->src(), send_from_host_node); in ReplaceRetNodesWithSendFromHostNode()
267 g->AddEdge(edge->src(), edge->src_output(), send_from_host_node, index); in ReplaceRetNodesWithSendFromHostNode()
272 g->AddEdge(key_placeholder, 0, send_from_host_node, in ReplaceRetNodesWithSendFromHostNode()
274 return send_from_host_node; in ReplaceRetNodesWithSendFromHostNode()
280 int num_inputs, Node* send_from_host_node) { in GetInferredInputShapes() argument
284 if (!send_from_host_node->input_edge(i, &e).ok()) { in GetInferredInputShapes()
2231 Node * send_from_host_node, in operator ()()
[all …]
|