Searched refs:worker_index (Results 1 – 1 of 1) sorted by relevance
86 def compute_devices_for_worker(self, worker_index): argument87 return self._fed_devices[worker_index]152 def out_of_range_fn(worker_index, device): argument156 data = self._iterators[worker_index].get_next(device)