Searched refs:reference_dims (Results 1 – 2 of 2) sorted by relevance
338 absl::Span<const int64> reference_dims);
1550 absl::Span<const int64> reference_dims) { in AlignShardingOnDims() argument1552 auto reference_grouped = GroupShardingOnDims(reference, reference_dims); in AlignShardingOnDims()