Searched refs:other_grouped (Results 1 – 1 of 1) sorted by relevance
2018 absl::optional<GroupedSharding> other_grouped = in PartitionDotGroupOnNonContracting() local2029 if (!other_grouped) { in PartitionDotGroupOnNonContracting()2043 if (other_grouped && other_grouped->group_dims.size() == 1 && in PartitionDotGroupOnNonContracting()2044 other_grouped->group_dims[0] == other.base_shape().rank()) { in PartitionDotGroupOnNonContracting()2046 other = other.Reshard(UngroupSharding(*other_grouped)); in PartitionDotGroupOnNonContracting()2049 partially_replicated_other->set_sharding(other_grouped->sharding); in PartitionDotGroupOnNonContracting()2051 other = other.Reshard(UngroupSharding(*other_grouped)); in PartitionDotGroupOnNonContracting()2055 other.sharding(), other_grouped->group_dims)) in PartitionDotGroupOnNonContracting()2059 partially_replicated_other->set_sharding(other_grouped->sharding); in PartitionDotGroupOnNonContracting()2377 absl::optional<GroupedSharding> other_grouped = in LhsIsBestMatchForNonContractingPartitioning() local[all …]