Searched refs:all_local_devices (Results 1 – 5 of 5) sorted by relevance
29 all_local_devices = mirrored_strategy.all_local_devices variable171 devices = mirrored_strategy.all_local_devices(num_gpus_per_worker)
292 return strategy_cls(mirrored_strategy.all_local_devices())
1436 mirrored_strategy.all_local_devices()),1546 mirrored_strategy.all_local_devices())1571 mirrored_strategy.all_local_devices())
254 return strategy_cls(mirrored_strategy.all_local_devices())
405 def all_local_devices(num_gpus=None): function418 return devices if devices else all_local_devices()