Searched refs:_split_by_task (Results 1 – 1 of 1) sorted by relevance
642 def _split_by_task(devices, values): function726 per_worker_devices, per_worker_values = _split_by_task(devices, input_tensors)817 per_worker_devices, per_worker_values = _split_by_task(devices, input_tensors)