Searched refs:main_graph (Results 1 – 1 of 1) sorted by relevance
1119 Status ExpandHostGraphIntoMainGraph(Graph* main_graph, in ExpandHostGraphIntoMainGraph() argument1149 node_map[host_graph->source_node()] = main_graph->source_node(); in ExpandHostGraphIntoMainGraph()1151 node_map[host_graph->sink_node()] = main_graph->sink_node(); in ExpandHostGraphIntoMainGraph()1165 copy = main_graph->AddNode(copy_def, &s); in ExpandHostGraphIntoMainGraph()1180 main_graph->AddEdge(node_map[e->src()], e->src_output(), copy, in ExpandHostGraphIntoMainGraph()1187 main_graph->AddControlEdge(copy, xla_computation_node); in ExpandHostGraphIntoMainGraph()