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