Searched refs:old_dim (Results 1 – 2 of 2) sorted by relevance
103 int old_dim = dims(); in Extend() local104 DCHECK_LE(old_dim, dim); in Extend()107 for (int d = old_dim; d < dim; ++d) { in Extend()
94 int64 old_dim = old.tile_assignment().dim(i); in MergeSharding() local96 new_tile_dims.push_back(old_dim); in MergeSharding()97 } else if (old_dim == 1) { in MergeSharding()99 } else if (new_dim == old_dim) { in MergeSharding()