Searched refs:lhs_tile_shape (Results 1 – 1 of 1) sorted by relevance
2100 auto lhs_tile_shape = lhs_sharding.tile_assignment().dimensions(); in PartitionDotGroupOnContracting() local2113 rhs_tile_shape[dim.rhs] = lhs_tile_shape[dim.lhs]; in PartitionDotGroupOnContracting()2122 lhs_tile_shape[dim.lhs] = rhs_tile_shape[dim.rhs]; in PartitionDotGroupOnContracting()2125 new_tile.Reshape(lhs_tile_shape); in PartitionDotGroupOnContracting()