Searched refs:dim_after_reshape (Results 1 – 1 of 1) sorted by relevance
2912 int64 dim_after_reshape = i >= split_dimension ? i + 1 : i; in AllToAllArray() local2916 permutation.push_back(dim_after_reshape); in AllToAllArray()