Searched refs:partially_replicated_other (Results 1 – 1 of 1) sorted by relevance
2042 auto partially_replicated_other = other.hlo(); in PartitionDotGroupOnNonContracting() local2047 partially_replicated_other = other.hlo(); in PartitionDotGroupOnNonContracting()2049 partially_replicated_other->set_sharding(other_grouped->sharding); in PartitionDotGroupOnNonContracting()2052 partially_replicated_other = in PartitionDotGroupOnNonContracting()2058 partially_replicated_other, partially_replicated_other->sharding()); in PartitionDotGroupOnNonContracting()2059 partially_replicated_other->set_sharding(other_grouped->sharding); in PartitionDotGroupOnNonContracting()2061 auto other_p = PartitionedHlo(partially_replicated_other, other.base_shape(), in PartitionDotGroupOnNonContracting()