Searched defs:dst_dim (Results 1 – 2 of 2) sorted by relevance
165 for (auto dst_dim : dst_sparse_dims) { in DenseToSparse() local
123 int64_t dst_dim = dst->tile_assignment().dim(i); in MergeShardingIfCompatible() local