Searched defs:upper_level_f (Results 1 – 1 of 1) sorted by relevance
710 def _build_nccl_hybrid(input_tensors, red_op, upper_level_f): argument779 def upper_level_f(x): function786 upper_level_f = lambda x: build_recursive_hd_all_reduce(x, red_op, un_op) function793 def upper_level_f(x): function799 def _build_shuffle_hybrid(input_tensors, gather_devices, red_op, upper_level_f): argument847 def upper_level_f(tensors): function859 def upper_level_f(tensors): function