Searched refs:_build_shuffle_hybrid (Results 1 – 1 of 1) sorted by relevance
797 def _build_shuffle_hybrid(input_tensors, gather_devices, red_op, upper_level_f): function847 return _build_shuffle_hybrid(859 return _build_shuffle_hybrid(