Fix conversions between ShardingParam and HloSharding.
Fix conversions between ShardingParam and HloSharding.

This change ensures that conversions work correctly for meshes with more than 2 axis, and adds additional tests for conversions between HloSharding to ShardingParam, ShardingParam to OpSharding, and ShardingParam to HloSharding.
