Searched refs:_split_by_task (Results 1 – 1 of 1) sorted by relevance
648 def _split_by_task(devices, values): function732 per_worker_devices, per_worker_values = _split_by_task(devices, input_tensors)823 per_worker_devices, per_worker_values = _split_by_task(devices, input_tensors)