Searched refs:tgt_dims_skipping_new (Results 1 – 1 of 1) sorted by relevance
1221 std::vector<int64> tgt_dims_skipping_new(tgt_to_src.size(), -1); in TransposeShardingWithCollapsedDims() local1227 tgt_dims_skipping_new[i] = i - skipped_tgt_dims; in TransposeShardingWithCollapsedDims()1240 perm[tgt_dims_skipping_new[src_to_tgt[i]]] = i; in TransposeShardingWithCollapsedDims()1248 tgt_tiles[i] = reshape_tiles.dim(tgt_dims_skipping_new[i]); in TransposeShardingWithCollapsedDims()