Searched refs:shuffle_red_op (Results 1 – 1 of 1) sorted by relevance
795 shuffle_red_op, un_op=None): argument798 return build_shuffle_all_reduce(x, gather_devices, shuffle_red_op, un_op)