Searched refs:compute_devices_for_worker (Results 1 – 1 of 1) sorted by relevance
86 def compute_devices_for_worker(self, worker_index): member in InputWorkers164 with ops.device(self._input_workers.compute_devices_for_worker(0)[0]):176 devices = self._input_workers.compute_devices_for_worker(i)266 devices = input_workers.compute_devices_for_worker(i)315 worker_devices = input_workers.compute_devices_for_worker(i)