When an HLO module is manually partitioned, infer a mesh shape to use rather than using the provided one. This wiill throw an error if auto-sharding infers multiple mesh shapes, but try_multiple_mesh_shapes=true.
When an HLO module is manually partitioned, infer a mesh shape to use rather than using the provided one. This wiill throw an error if auto-sharding infers multiple mesh shapes, but try_multiple_mesh_shapes=true.
