Searched refs:_build_nccl_hybrid (Results 1 – 1 of 1) sorted by relevance
714 def _build_nccl_hybrid(input_tensors, red_op, upper_level_f): function785 return _build_nccl_hybrid(input_tensors, red_op, upper_level_f)791 return _build_nccl_hybrid(input_tensors, red_op, upper_level_f)800 return _build_nccl_hybrid(input_tensors, nccl_red_op, upper_level_f)