xla

[LoweringContext] Support an optimized parameter mapping for SPMD

#8460

Merged