Add the result sharding to reshapes added in `ReshapeWithCorrectRepresentationAndSharding`.
Add the result sharding to reshapes added in `ReshapeWithCorrectRepresentationAndSharding`.

If `ConvertMlirHloToHlo` is applied after sharding propagation, then it's important that the added reshapes will have the same sharding as that of the corresponding function outputs.
