xla

022ecde6 - Migrate runtime.xla_device in favor of core.xla_model.xla_device (#9200)