xla

dc057dcd - Enable SPMD + dynamo for inference (#5002)

Enable SPMD + dynamo for inference (#5002) * initial code change * Add simple test, which currently failed * Add try catch to dynamo, use SPMD device in dynamo, create sharded placeholder if SPMD is enabled * fix review comments * change comments