Searched refs:perm_dimensions (Results 1 – 1 of 1) sorted by relevance
273 auto perm_dimensions = dimensions; in TransposeSharding() local276 perm_dimensions.push_back(dimensions.size()); in TransposeSharding()278 const int64 rank = perm_dimensions.size(); in TransposeSharding()281 tile_assignment_dim[i] = sharding.tile_assignment().dim(perm_dimensions[i]); in TransposeSharding()288 src_indices[perm_dimensions[i]] = indices[i]; in TransposeSharding()