Searched refs:lhs_partitions (Results 1 – 1 of 1) sorted by relevance
2664 int64 lhs_partitions = lhs.sharding().tile_assignment().dim(dim.lhs); in PartitionDot() local2665 if (lhs_partitions > 1 && in PartitionDot()2666 lhs_partitions == output_sharding.tile_assignment().dim(dim.output)) { in PartitionDot()2735 int64 lhs_partitions = lhs.sharding().tile_assignment().dim(dim.lhs); in PartitionDot() local2736 if (lhs_partitions > 1 && in PartitionDot()2737 lhs_partitions == rhs.sharding().tile_assignment().dim(dim.rhs)) { in PartitionDot()