Searched refs:send_node_in_host_graph (Results 1 – 1 of 1) sorted by relevance
1233 Node* send_node_in_host_graph = nullptr; in RewriteShapeInferenceGraph() local1236 send_node_in_host_graph = n; in RewriteShapeInferenceGraph()1240 if (send_node_in_host_graph) { in RewriteShapeInferenceGraph()1256 std::vector<Visit> stack{{send_node_in_host_graph, false}}; in RewriteShapeInferenceGraph()1291 send_from_host = node_map[send_node_in_host_graph]; in RewriteShapeInferenceGraph()