Lines Matching refs:local_devices
119 local_devices = tuple("/device:GPU:%d" % i for i in range(num_gpus))
121 local_devices = ("/device:CPU:0",)
126 super(CollectiveAllReduceExtended, self)._initialize_local(local_devices)
147 logging.info("CollectiveAllReduceStrategy with local_devices = %r",
148 local_devices)
185 local_devices = tuple("%s/device:GPU:%d" % (self._worker_device, i)
188 local_devices = (self._worker_device,)
191 super(CollectiveAllReduceExtended, self)._initialize_local(local_devices)
213 "task_type = %r, task_id = %r, num_workers = %r, local_devices = %r, "
215 task_id, self._num_workers, local_devices,
