Searched defs:upper_level_f (Results 1 – 1 of 1) sorted by relevance
708 def _build_nccl_hybrid(input_tensors, red_op, upper_level_f): argument777 def upper_level_f(x): function784 upper_level_f = lambda x: build_recursive_hd_all_reduce(x, red_op, un_op) function791 def upper_level_f(x): function797 def _build_shuffle_hybrid(input_tensors, gather_devices, red_op, upper_level_f): argument845 def upper_level_f(tensors): function857 def upper_level_f(tensors): function