experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[XLA:Python] Make sure xla_client is always imported before TPU client extension.

Browse Source 
PiperOrigin-RevId: 311144400
Change-Id: Ia499185c36b5596b7aa25c44e51fd07696f85cfe
This commit is contained in:
Skye Wanderman-Milne 2020-05-12 09:46:28 -07:00 committed by TensorFlower Gardener
parent 3116ec3708
commit f462867806
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/compiler/xla/python/tpu_driver/client/tpu_client.py
View File

@ -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