Searched refs:BuildFullMeshDeviceAssignment (Results 1 – 1 of 1) sorted by relevance
1502 static Status BuildFullMeshDeviceAssignment( in BuildFullMeshDeviceAssignment() function1639 return BuildFullMeshDeviceAssignment(num_replicas, tpu_devices, num_tasks, in BuildDeviceAssignment()