Searched refs:xla_computation_node (Results 1 – 2 of 2) sorted by relevance
517 Node* xla_computation_node; in Encapsulate() local520 xla_computation_node = n; in Encapsulate()523 if (!xla_computation_node) { in Encapsulate()529 XlaClusterInfo{func, func_name_attrs, xla_computation_node, in Encapsulate()
1122 Node* xla_computation_node, in ExpandHostGraphIntoMainGraph() argument1187 main_graph->AddControlEdge(copy, xla_computation_node); in ExpandHostGraphIntoMainGraph()