xla

7008ac11 - Add the initial support for local vs. remote devices, to support feeding slices bigger than single TPU host.