Searched refs:dim_permutation (Results 1 – 1 of 1) sorted by relevance
1160 std::vector<int64> dim_permutation( in PartiallyReplicateTiledShardingOnDims() local1162 std::iota(dim_permutation.begin(), dim_permutation.end(), 0); in PartiallyReplicateTiledShardingOnDims()1163 absl::c_sort(dim_permutation, [&](const int64 a, const int64 b) { in PartiallyReplicateTiledShardingOnDims()1167 auto transposed = TransposeSharding(sharding, dim_permutation); in PartiallyReplicateTiledShardingOnDims()