xla

Support input sharding changed after first dynamo tracing

#5477

Merged