Searched refs:_build_shuffle_hybrid (Results 1 – 1 of 1) sorted by relevance
803 def _build_shuffle_hybrid(input_tensors, gather_devices, red_op, upper_level_f): function853 return _build_shuffle_hybrid(865 return _build_shuffle_hybrid(