Searched refs:_build_shuffle_scatter (Results 1 – 1 of 1) sorted by relevance
586 output_tensors = _build_shuffle_scatter(reduced_shards, dst_devices)629 def _build_shuffle_scatter(reduced_shards, dst_devices): function838 output_tensors += _build_shuffle_scatter(