Searched refs:input_workers (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib.py | 125 def __init__(self, input_workers, iterators): argument
126 assert isinstance(input_workers, InputWorkers)
127 if not input_workers.worker_devices:
131 self._input_workers = input_workers
239 def __init__(self, input_fn, input_workers, input_contexts): argument
254 assert isinstance(input_workers, InputWorkers)
255 if input_workers.num_workers != len(input_contexts):
259 (input_workers.num_workers, len(input_contexts)))
263 worker = input_workers.worker_devices[i]
266 devices = input_workers.compute_devices_for_worker(i)
[all …]
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | input_lib_test.py | 41 input_workers = input_lib.InputWorkers(device_map, worker_device_pairs)
45 for i in range(input_workers.num_workers):
48 num_input_pipelines=input_workers.num_workers,
52 iterator = input_lib.InputFunctionIterator(dataset_fn, input_workers,
56 dataset_fn(distribute_lib.InputContext()), input_workers,
|