Home
last modified time | relevance | path

Searched refs:_num_workers (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dcollective_all_reduce_strategy.py195 "num_workers").set(self.extended._num_workers)
294 "num_workers").set(self.extended._num_workers)
343 self._num_workers = 1
381 group_size=self._num_workers,
422 self._num_workers = multi_worker_util.worker_count(cluster_spec, task_type)
423 if not self._num_workers:
491 group_size=len(local_devices) * self._num_workers,
496 group_size=self._num_workers,
517 self._num_workers, local_devices,
549 group_size = self._num_workers
[all …]
Dparameter_server_strategy_v2.py454 self._num_workers = len(cluster_spec.as_dict().get("worker", ()))
459 for i in range(self._num_workers):
481 "ps_strategy_num_workers").set(self._num_workers)
Ddistribute_coordinator.py145 self._num_workers = _get_num_workers(cluster_spec)
308 return self._num_workers
Dcollective_all_reduce_strategy_test.py570 strategy.extended._num_workers, results[1].numpy())
/external/tensorflow/tensorflow/python/data/experimental/service/
Dserver_lib_test.py158 self.assertEqual(0, dispatcher._num_workers())
161 self.assertEqual(1, dispatcher._num_workers())
164 self.assertEqual(2, dispatcher._num_workers())
Dserver_lib.py212 def _num_workers(self): member in DispatchServer
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Ddata_service_test_base.py154 return self.dispatcher._num_workers()
/external/tensorflow/tensorflow/python/distribute/coordinator/
Dcluster_coordinator.py870 self._num_workers = strategy._num_workers
893 "/job:worker/replica:0/task:%d" % i for i in range(self._num_workers)