Searched refs:build_recursive_hd_all_reduce (Results 1 – 2 of 2) sorted by relevance
426 def build_recursive_hd_all_reduce(input_tensors, red_op, un_op=None): function784 upper_level_f = lambda x: build_recursive_hd_all_reduce(x, red_op, un_op)
218 return lambda x, un_op: ar.build_recursive_hd_all_reduce(