Searched refs:local_devices_from_num_gpus (Results 1 – 3 of 3) sorted by relevance
75 return cls(device_util.local_devices_from_num_gpus(num_gpus))
130 def local_devices_from_num_gpus(num_gpus): function
300 compute_devices = device_util.local_devices_from_num_gpus(num_gpus)