Searched refs:worker_device_pairs (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | input_lib_test.py | 38 def _create_iterator(self, input_type, dataset_fn, worker_device_pairs, argument
41 input_workers = input_lib.InputWorkers(device_map, worker_device_pairs)
63 worker_device_pairs, argument
67 devices = nest.flatten([ds for _, ds in worker_device_pairs])
69 input_type, dataset_fn, worker_device_pairs, devices, split_batch_by)
106 worker_device_pairs = [("", ["/device:CPU:0"])]
111 self._test_iterator(input_type, dataset_fn, worker_device_pairs,
119 worker_device_pairs = [("", ["/device:GPU:0", "/device:CPU:0"])]
124 self._test_iterator(input_type, dataset_fn, worker_device_pairs,
132 worker_device_pairs = [("", ["/device:GPU:0", "/device:CPU:0"])]
[all …]
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib.py | 44 def __init__(self, device_map, worker_device_pairs=None, logical_device=0): argument
56 if worker_device_pairs is None:
57 worker_device_pairs = ((
60 self._input_worker_devices = tuple(d for d, _ in worker_device_pairs)
62 for _, f in worker_device_pairs)
|
| D | one_device_strategy.py | 56 worker_device_pairs = [(self._input_device, [self._device])]
59 device_map, worker_device_pairs)
|