Searched refs:all_to_all (Results 1 – 1 of 1) sorted by relevance
44 def all_to_all(x, function68 return gen_tpu_ops.all_to_all(83 gen_tpu_ops.all_to_all(