Searched refs:build_nccl_all_reduce (Results 1 – 1 of 1) sorted by relevance
685 def build_nccl_all_reduce(input_tensors, red_op, un_op=None): function739 worker_values = build_nccl_all_reduce(per_worker_values[w], red_op)