Searched refs:build_nccl_then_recursive_hd (Results 1 – 2 of 2) sorted by relevance
399 summed_grads = all_reduce.build_nccl_then_recursive_hd(
782 def build_nccl_then_recursive_hd(input_tensors, red_op, un_op=None): function