Home
last modified time | relevance | path

Searched refs:_input_workers (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dinput_lib.py546 …for i, worker in enumerate(iterator._input_workers.worker_devices): # pylint: disable=protected-a…
622 self._input_workers = input_workers
653 for i, worker in enumerate(self._input_workers.worker_devices):
677 for i, worker in enumerate(self._input_workers.worker_devices):
679 devices = self._input_workers.compute_devices_for_worker(i)
744 for i, w in enumerate(self._input_workers.worker_devices):
772 self._input_workers = input_workers
785 return (self._input_workers.serialize(), self._element_spec,
807 if self._input_workers.serialize() != other._input_workers.serialize():
816 return DistributedIteratorSpec(self._input_workers, element_spec,
[all …]
Dinput_lib_type_spec_test.py76 self.assertEqual(spec._input_workers, iterator._input_workers)
113 self.assertEqual(iterator._input_workers, re_iterator._input_workers)
320 self.assertEqual(iterator._input_workers, re_iterator._input_workers)
368 self.assertEqual(spec._input_workers, iterator._input_workers)
443 self.assertEqual(iterator._input_workers, re_iterator._input_workers)
Done_device_strategy.py271 def _input_workers(self): member in OneDeviceExtended
293 return input_lib.DatasetIterator(dataset, self._input_workers,
300 return input_lib.InputFunctionIterator(input_fn, self._input_workers,
Dmirrored_strategy.py465 def _input_workers(self): member in MirroredExtended
540 self._input_workers,
549 num_workers = self._input_workers.num_workers
555 return input_lib.InputFunctionIterator(input_fn, self._input_workers,
Dparameter_server_strategy.py335 def _input_workers(self): member in ParameterServerStrategyExtended
351 self._input_workers,
372 return input_lib.InputFunctionIterator(input_fn, self._input_workers,
Dcollective_all_reduce_strategy.py534 def _input_workers(self): member in CollectiveAllReduceExtended
640 self._input_workers,
651 return input_lib.InputFunctionIterator(input_fn, self._input_workers,
Ddistribute_lib_test.py72 self._input_workers = input_lib.InputWorkers(worker_device_pairs)
87 return input_lib.InputFunctionIterator(input_fn, self._input_workers,
Dinput_lib_test.py1062 distribution.extended._input_workers,
/external/tensorflow/tensorflow/python/distribute/coordinator/
Dcluster_coordinator.py1326 self._input_workers = input_workers