Searched refs:tgt_tiles (Results 1 – 1 of 1) sorted by relevance
1245 std::vector<int64> tgt_tiles(tgt_to_src.size(), 1); in TransposeShardingWithCollapsedDims() local1246 for (int64 i = 0; i < tgt_tiles.size(); ++i) { in TransposeShardingWithCollapsedDims()1248 tgt_tiles[i] = reshape_tiles.dim(tgt_dims_skipping_new[i]); in TransposeShardingWithCollapsedDims()1251 reshape_tiles.Reshape(tgt_tiles); in TransposeShardingWithCollapsedDims()