Searched refs:nccl_red_op (Results 1 – 1 of 1) sorted by relevance
788 def build_nccl_then_shuffle(input_tensors, gather_devices, nccl_red_op, argument794 return _build_nccl_hybrid(input_tensors, nccl_red_op, upper_level_f)