Searched refs:experimental_device_assignment (Results 1 – 3 of 3) sorted by relevance
204 resolver, experimental_device_assignment=device_assignment)211 resolver, experimental_device_assignment=device_assignment2)322 resolver, experimental_device_assignment=device_assignment)330 resolver, experimental_device_assignment=device_assignment2)357 resolver, experimental_device_assignment=device_assignment)365 resolver, experimental_device_assignment=device_assignment2)1108 experimental_device_assignment=device_assignment)1177 experimental_device_assignment=device_assignment)1225 experimental_device_assignment=device_assignment)1271 experimental_device_assignment=device_assignment)
321 experimental_device_assignment=None): argument335 device_assignment=experimental_device_assignment))
21 …argspec: "args=[\'self\', \'tpu_cluster_resolver\', \'experimental_device_assignment\'], varargs=N…