Searched refs:worker_device_pairs (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib_test.py | 167 worker_device_pairs, argument
182 devices = nest.flatten([ds for _, ds in worker_device_pairs])
183 input_workers = input_lib.InputWorkers(worker_device_pairs)
300 worker_device_pairs = [("/device:CPU:0", ["/device:CPU:0"])]
301 input_workers = input_lib.InputWorkers(worker_device_pairs)
306 input_workers = input_lib.InputWorkers(worker_device_pairs)
338 worker_device_pairs = [("/device:CPU:0", ["/device:GPU:0",
342 input_workers = input_lib.InputWorkers(worker_device_pairs)
368 worker_device_pairs = [("/device:CPU:0", ["/device:CPU:0"])]
378 dataset_or_input_fn, worker_device_pairs,
[all …]
|
| D | input_lib.py | 492 def __init__(self, worker_device_pairs): argument
499 self._worker_device_pairs = worker_device_pairs
525 def deserialize(self, worker_device_pairs): argument
526 return InputWorkers(worker_device_pairs)
824 … worker_device_pairs = self._input_workers._worker_device_pairs # pylint: disable=protected-access
826 for i, (input_device, compute_devices) in enumerate(worker_device_pairs):
|
| D | distribute_lib_test.py | 71 worker_device_pairs = [("", ["/device:CPU:0"])]
72 self._input_workers = input_lib.InputWorkers(worker_device_pairs)
|