Searched refs:global_tpu_array_node (Results 1 – 1 of 1) sorted by relevance
123 Graph* graph, Node** global_tpu_array_node) { in AddGlobalTPUArrayNode() argument130 TF_ASSIGN_OR_RETURN(*global_tpu_array_node, in AddGlobalTPUArrayNode()132 (*global_tpu_array_node)->set_assigned_device_name(host_device_name); in AddGlobalTPUArrayNode()133 graph->AddEdge(wait_node, 0, *global_tpu_array_node, 0); in AddGlobalTPUArrayNode()