message: [XLA:Python] Make sure xla_client is always imported before TPU client extension.
diff: @@ -20,6 +20,9 @@ from __future__ import print_function
  from absl import logging
  +# Import xla_client to load shared C++ extensions (just CompileOptions at the
  +# time of writing).
  +from tensorflow.compiler.xla.python import xla_client # pylint: disable=unused-import
  from tensorflow.compiler.xla.python.tpu_driver.client import tpu_client_extension as _tpu_client
  