This CL adds support for handling the SPMDFullToShardShape and SPMDShardToFullShape custom calls in auto-sharding.
This CL adds support for handling the SPMDFullToShardShape and SPMDShardToFullShape custom calls in auto-sharding.
