Searched refs:rhs_partitions (Results 1 – 1 of 1) sorted by relevance
2691 int64 rhs_partitions = rhs.sharding().tile_assignment().dim(dim.rhs); in PartitionDot() local2692 if (rhs_partitions > 1 && in PartitionDot()2693 rhs_partitions == output_sharding.tile_assignment().dim(dim.output)) { in PartitionDot()