Searched refs:_build_nccl_hybrid (Results 1 – 1 of 1) sorted by relevance
708 def _build_nccl_hybrid(input_tensors, red_op, upper_level_f): function779 return _build_nccl_hybrid(input_tensors, red_op, upper_level_f)785 return _build_nccl_hybrid(input_tensors, red_op, upper_level_f)794 return _build_nccl_hybrid(input_tensors, nccl_red_op, upper_level_f)