Searched refs:_all_to_all (Results 1 – 1 of 1) sorted by relevance
1559 def _all_to_all(value, axis): function1608 ys = [_all_to_all(t, axis=axis) for t in nest.flatten(value)]