message: Fix the devices order in TPUStrategy.
diff: @@ -233,7 +233,9 @@ class TPUExtended(distribute_lib.DistributionStrategyExtended):
  }
  self._host_device = tpu_strategy_util.get_first_tpu_host_device(
  self._tpu_cluster_resolver)
  - self._tpu_devices = tuple(sorted(self._device_index.keys()))
  + # We sort the devices by the indexes in tpu_metadata.devices.
  + self._tpu_devices = tuple(device[0] for device in sorted(
  + self._device_index.items(), key=lambda device: device[1]))
  # Only create variables for the number of replicas we're running.
  self._tpu_devices = self._tpu_devices[:self._num_replicas_in_sync]
  self._device_map = values.ReplicaDeviceMap(self._tpu_devices)
  