Searched defs:lhs_partitions (Results 1 – 1 of 1) sorted by relevance
2664 int64 lhs_partitions = lhs.sharding().tile_assignment().dim(dim.lhs); in PartitionDot() local2735 int64 lhs_partitions = lhs.sharding().tile_assignment().dim(dim.lhs); in PartitionDot() local