Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/distribute/
Done_device_strategy.py265 return input_lib.InputWorkers([(self._input_device, (self._device,))])
267 return input_lib.InputWorkers([(self._input_device,
Dmirrored_strategy.py443 return input_lib.InputWorkers(self._input_workers_devices)
454 return input_lib.InputWorkers(self._input_workers_devices)
457 return input_lib.InputWorkers([
462 return input_lib.InputWorkers(self._input_workers_devices)
Dinput_lib.py489 class InputWorkers(object): class
526 return InputWorkers(worker_device_pairs)
617 assert isinstance(input_workers, InputWorkers)
930 assert isinstance(input_workers, InputWorkers)
1392 assert isinstance(input_workers, InputWorkers)
1896 assert isinstance(input_workers, InputWorkers)
Dinput_lib_test.py183 input_workers = input_lib.InputWorkers(worker_device_pairs)
301 input_workers = input_lib.InputWorkers(worker_device_pairs)
306 input_workers = input_lib.InputWorkers(worker_device_pairs)
342 input_workers = input_lib.InputWorkers(worker_device_pairs)
556 input_workers = input_lib.InputWorkers(worker_device_pairs)
1682 input_workers = input_lib.InputWorkers(worker_device_pairs)
Dtpu_strategy.py879 input_workers = input_lib.InputWorkers(
892 input_workers = input_lib.InputWorkers(
913 return input_lib.InputWorkers(
916 return input_lib.InputWorkers(
Dparameter_server_strategy.py327 return input_lib.InputWorkers(
330 return input_lib.InputWorkers(
Dcollective_all_reduce_strategy.py526 return input_lib.InputWorkers([(host_device, self.worker_devices)])
528 return input_lib.InputWorkers([(
Ddistribute_lib_test.py72 self._input_workers = input_lib.InputWorkers(worker_device_pairs)
/external/tensorflow/tensorflow/python/distribute/coordinator/
Dcluster_coordinator.py1217 input_workers = input_lib.InputWorkers([