| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib.py | 509 def worker_devices(self): member in InputWorkers
516 devices = self.worker_devices
545 worker_devices = []
546 …for i, worker in enumerate(iterator._input_workers.worker_devices): # pylint: disable=protected-a…
554 worker_devices.append(worker_has_value.device)
618 if not input_workers.worker_devices:
653 for i, worker in enumerate(self._input_workers.worker_devices):
677 for i, worker in enumerate(self._input_workers.worker_devices):
744 for i, w in enumerate(self._input_workers.worker_devices):
1002 input_workers.worker_devices)
[all …]
|
| D | collective_all_reduce_strategy.py | 526 return input_lib.InputWorkers([(host_device, self.worker_devices)])
531 self.worker_devices])])
626 num_local_replicas = len(self.worker_devices)
737 if num_devices == len(self.worker_devices):
756 len(self.worker_devices) == 1):
768 reduce_op, value, destinations, len(self.worker_devices))
916 return len(self.worker_devices) * self._num_workers
931 return self._id_in_cluster * len(self.worker_devices) + replica_id
935 self._id_in_cluster * len(self.worker_devices))
|
| D | one_device_strategy_test.py | 77 iterator, distribution.extended.worker_devices, expected_values)
92 iterator, distribution.extended.worker_devices, expected_values,
140 distribution.extended.worker_devices[0]).device_type)
|
| D | tf_function_test.py | 57 worker = distribution.extended.worker_devices[0]
83 worker = distribution.extended.worker_devices[0]
|
| D | mirrored_strategy_test.py | 183 self._test_input_fn_iterator(iterator, distribution.extended.worker_devices,
200 self._test_input_fn_iterator(iterator, distribution.extended.worker_devices,
256 device in distribution.extended.worker_devices]
790 distribution.extended.worker_devices[0], mirrored_var._devices[0])
792 distribution.extended.worker_devices[1], mirrored_var._devices[1])
805 distribution.extended.worker_devices[0], mirrored_var._devices[0])
807 distribution.extended.worker_devices[1], mirrored_var._devices[1])
868 distribution.extended.worker_devices[0], mirrored_var._devices[0])
870 distribution.extended.worker_devices[1], mirrored_var._devices[1])
1207 iterator, distribution.extended.worker_devices, expected_values, sess)
[all …]
|
| D | tpu_strategy_test.py | 369 self.assertLen(first_core_strategy.extended.worker_devices, 1)
370 self.assertEndsWith(first_core_strategy.extended.worker_devices[0],
373 self.assertLen(second_core_strategy.extended.worker_devices, 1)
374 self.assertEndsWith(second_core_strategy.extended.worker_devices[0],
859 for device in set(strategy.extended.worker_devices):
865 self.assertLen(strategy.extended.worker_devices, trace_count[0])
1014 distribution.extended.worker_devices,
1252 strategy0.extended.worker_devices)
1275 strategy1.extended.worker_devices)
|
| D | mirrored_strategy.py | 404 worker_devices = []
409 worker_devices.append((worker, device_dict[job][task]))
422 self._input_workers_devices = worker_devices
809 def worker_devices(self): member in MirroredExtended
818 return self.worker_devices
|
| D | strategy_test_lib.py | 288 expected_devices = [False] * len(d.extended.worker_devices)
293 self.assertLess(replica_id, len(d.extended.worker_devices))
299 [True] * len(d.extended.worker_devices))
365 self._test_input_fn_iterator(iterator, strategy.extended.worker_devices,
771 iterator, distribution.extended.worker_devices, expected_values, sess)
796 iterator, distribution.extended.worker_devices, expected_values, sess,
|
| D | one_device_strategy.py | 437 def worker_devices(self): member in OneDeviceExtended
487 return self._strategy.extended.worker_devices
|
| D | values_test.py | 88 devices = distribution.extended.worker_devices
294 for i in range(len(distribution.extended.worker_devices)):
310 worker_devices = distribution.extended.worker_devices
314 worker_device_id = ctx.replica_id_in_sync_group % len(worker_devices)
315 with ops.device(worker_devices[worker_device_id]):
319 for i in range(len(distribution.extended.worker_devices)):
321 worker_devices[i])
1107 devices = strategy.extended.worker_devices
|
| D | input_lib_test.py | 452 for tpu_device in distribution.extended.worker_devices:
905 for replica_id in range(len(distribution.extended.worker_devices)):
1063 len(distribution.extended.worker_devices),
1495 assert x.values[0].device == distribution.extended.worker_devices[0]
1496 assert x.values[0].backing_device == distribution.extended.worker_devices[
1498 assert x.values[1].device == distribution.extended.worker_devices[1]
1499 assert x.values[1].backing_device == distribution.extended.worker_devices[
|
| D | test_util.py | 62 assert len(strategy.extended.worker_devices) == len(value._values)
|
| D | custom_training_loop_input_test.py | 416 self.assertAllEqual(backing_devices, distribution.extended.worker_devices)
433 self.assertAllEqual(backing_devices, distribution.extended.worker_devices)
450 self.assertAllEqual(backing_devices, distribution.extended.worker_devices)
|
| D | mirrored_variable_test.py | 497 for i, _ in enumerate(distribution.extended.worker_devices):
509 expected_mean /= len(distribution.extended.worker_devices)
|
| D | collective_all_reduce_strategy_test.py | 223 devices = distribution.extended.worker_devices
569 list(range(len(strategy.extended.worker_devices))) *
|
| D | tpu_strategy.py | 1380 def worker_devices(self): member in TPUExtended
1385 return self.worker_devices
1546 return (ds.extended.worker_devices[replica_id],)
|
| D | distribute_lib.py | 789 len(extended.worker_devices) > 1)
2584 def worker_devices(self): member in StrategyExtendedV2
3503 def worker_devices(self): member in _DefaultDistributionExtended
|
| D | mirrored_run.py | 151 devices = distribution.extended.worker_devices
|
| D | parameter_server_strategy.py | 658 def worker_devices(self): member in ParameterServerStrategyExtended
|
| D | parameter_server_strategy_test.py | 532 devices = distribution.extended.worker_devices
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 14 name: "worker_devices"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-strategy-extended.pbtxt | 31 name: "worker_devices"
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | iterator_test.py | 673 worker_devices = [
677 for device in worker_devices:
683 targets = dataset_ops.Dataset.from_tensor_slices(worker_devices)
704 expected_values = worker_devices
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_v2.py | 1304 tpu_device = self._strategy.extended.worker_devices[replica_id]
1655 for tpu_device in strategy.extended.worker_devices:
|
| D | tpu_embedding_v2_test.py | 754 device = strategy.extended.worker_devices[core_id]
|