Searched defs:upper_level_f (Results 1 – 1 of 1) sorted by relevance
714 def _build_nccl_hybrid(input_tensors, red_op, upper_level_f): argument783 def upper_level_f(x): function790 upper_level_f = lambda x: build_recursive_hd_all_reduce(x, red_op, un_op) function797 def upper_level_f(x): function803 def _build_shuffle_hybrid(input_tensors, gather_devices, red_op, upper_level_f): argument851 def upper_level_f(tensors): function863 def upper_level_f(tensors): function