Searched refs:send_from_host_node (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | extract_outside_compilation_pass.cc | 242 TF_ASSIGN_OR_RETURN(Node * send_from_host_node, in BuildSendFromHostNode()
244 return send_from_host_node; in BuildSendFromHostNode()
256 Node * send_from_host_node, in ReplaceRetNodesWithSendFromHostNode()
264 g->AddControlEdge(edge->src(), send_from_host_node); in ReplaceRetNodesWithSendFromHostNode()
266 g->AddEdge(edge->src(), edge->src_output(), send_from_host_node, index); in ReplaceRetNodesWithSendFromHostNode()
271 g->AddEdge(key_placeholder, 0, send_from_host_node, in ReplaceRetNodesWithSendFromHostNode()
273 return send_from_host_node; in ReplaceRetNodesWithSendFromHostNode()
279 int num_inputs, Node* send_from_host_node) { in GetInferredInputShapes() argument
283 if (!send_from_host_node->input_edge(i, &e).ok()) { in GetInferredInputShapes()
2232 Node * send_from_host_node, in operator ()()
[all …]
|