Searched refs:num_gpus_per_worker (Results 1 – 1 of 1) sorted by relevance
86 def _cluster_spec_to_device_list(cluster_spec, num_gpus_per_worker): argument92 if num_gpus_per_worker == 0:97 for gpu_id in range(num_gpus_per_worker)686 num_gpus_per_worker = _infer_num_gpus_per_worker(self._devices)688 cluster_spec, num_gpus_per_worker)