Searched refs:_build_shuffle_scatter (Results 1 – 1 of 1) sorted by relevance
580 output_tensors = _build_shuffle_scatter(reduced_shards, dst_devices)623 def _build_shuffle_scatter(reduced_shards, dst_devices): function832 output_tensors += _build_shuffle_scatter(