change xla_data_pb2 import

khcs · Nov 18, 2018 · 1d90243 · 1d90243
1 parent a30e858
commit 1d90243
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jax/lib/xla_bridge.py b/jax/lib/xla_bridge.py
@@ -29,7 +29,7 @@
 from absl import flags
 import numpy as onp  # 'onp' rather than 'np' to distinguish from autograd.numpy
 
-from tensorflow.compiler.xla import xla_data_pb2
+from . import xla_data_pb2
 from . import xla_client