Searched refs:collective_permute (Results 1 – 1 of 1) sorted by relevance
111 def collective_permute(x, source_target_pairs, name=None): function134 return gen_tpu_ops.collective_permute(x, source_target_pairs, name=name)143 return [gen_tpu_ops.collective_permute(grad, source_target_pairs), None]