| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | input_lib_test.py | 207 worker_devices = self._cpu_devices()
210 self._test_iterator(input_type, dataset_fn, worker_devices,
218 worker_devices = self._cpu_and_one_gpu_devices()
221 self._test_iterator(input_type, dataset_fn, worker_devices,
228 worker_devices = self._cpu_devices()
238 self._test_iterator(input_type, dataset_fn, worker_devices,
245 worker_devices = self._cpu_and_one_gpu_devices()
250 self._test_iterator(input_type, dataset_fn, worker_devices,
257 worker_devices = self._cpu_and_one_gpu_devices()
268 self._test_iterator(input_type, dataset_fn, worker_devices,
|
| D | one_device_strategy_test.py | 59 iterator, distribution.extended.worker_devices, expected_values)
74 iterator, distribution.extended.worker_devices, expected_values,
|
| D | cross_device_ops_test.py | 131 devices = distribution.extended.worker_devices
370 worker_devices = [ variable in MultiWorkerCrossDeviceOpsTest
378 worker_devices, 2, ("pscpu/pscpu", 2, -1), 0, 0, 0)),
382 worker_devices, 2, ("pscpu/pscpu", 2, -1), 1, 0, 0)),
386 worker_devices, 2, ("pscpu/pscpu", 2, -1), 0, 100, 10)),
390 worker_devices, 2, [("pscpu/pscpu", 2, 100),
|
| D | mirrored_strategy_multigpu_test.py | 119 self._test_input_fn_iterator(iterator, distribution.extended.worker_devices,
137 self._test_input_fn_iterator(iterator, distribution.extended.worker_devices,
708 for i, d in enumerate(distribution.extended.worker_devices):
718 expected_mean /= len(distribution.extended.worker_devices)
774 distribution.extended.worker_devices[0]).read_value()))
784 distribution.extended.worker_devices[0]).read_value()))
1489 iterator, distribution.extended.worker_devices, expected_values, sess)
1514 iterator, distribution.extended.worker_devices, expected_values, sess,
|
| D | strategy_test_lib.py | 252 expected_devices = [False] * len(d.extended.worker_devices)
257 self.assertLess(replica_id, len(d.extended.worker_devices))
263 [True] * len(d.extended.worker_devices))
|
| D | collective_all_reduce_strategy_test.py | 318 devices = distribution.extended.worker_devices
|
| D | parameter_server_strategy_test.py | 578 devices = distribution.extended.worker_devices
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib.py | 83 def worker_devices(self): member in InputWorkers
90 devices = self.worker_devices
127 if not input_workers.worker_devices:
137 for i, worker in enumerate(self._input_workers.worker_devices):
174 for i, worker in enumerate(self._input_workers.worker_devices):
230 for i, w in enumerate(self._input_workers.worker_devices):
263 worker = input_workers.worker_devices[i]
313 for i, worker in enumerate(input_workers.worker_devices):
315 worker_devices = input_workers.compute_devices_for_worker(i)
321 worker_devices)
|
| D | one_device_strategy.py | 179 def worker_devices(self): member in OneDeviceExtended
219 return self._strategy.extended.worker_devices
|
| D | collective_all_reduce_strategy.py | 193 self._device_map, [(self._worker_device, self.worker_devices)])
437 len(self.worker_devices) == 1):
471 return len(self.worker_devices) * self._num_workers
|
| D | mirrored_strategy.py | 489 worker_devices = []
494 worker_devices.append((worker, device_dict[job][task]))
508 self._device_map, worker_devices)
757 def worker_devices(self): member in MirroredExtended
|
| D | parameter_server_strategy.py | 378 (d, self._input_workers.worker_devices[0]))
517 def worker_devices(self): member in ParameterServerStrategyExtended
|
| D | tpu_strategy.py | 574 def worker_devices(self): member in TPUExtended
645 return (ds.extended.worker_devices[replica_id],)
|
| D | cross_device_ops.py | 810 worker_devices, argument
845 self._worker_devices = worker_devices
|
| D | distribute_lib.py | 1297 def worker_devices(self): member in DistributionStrategyExtended
1639 def worker_devices(self): member in _DefaultDistributionExtended
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 30 name: "worker_devices"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 30 name: "worker_devices"
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | iterator_test.py | 670 worker_devices = [
674 for device in worker_devices:
680 targets = dataset_ops.Dataset.from_tensor_slices(worker_devices)
701 expected_values = worker_devices
|