diff options
| author |  Justine Tunney <jart@google.com> | 2016-12-16 18:16:15 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-12-16 18:26:25 -0800 |
| commit | 58201a058853de647b37ddb0ccf63d89b2357f03 (patch) | |
| tree | a70f1626ef6c6bae985c3b581e2a6f9ce6f6b96e /tensorflow/contrib/compiler | |
| parent | 157dbdda6d7d6e4f679c5e89d0bd3e8c0a6085d5 (diff) | |
Remove hourglass imports from even more tests
Change: 142318245
Diffstat (limited to 'tensorflow/contrib/compiler')
| -rw-r--r-- | tensorflow/contrib/compiler/BUILD | 4 | ||||
| -rw-r--r-- | tensorflow/contrib/compiler/jit.py | 7 |
2 files changed, 8 insertions, 3 deletions
diff --git a/tensorflow/contrib/compiler/BUILD b/tensorflow/contrib/compiler/BUILD index 444f5e9e16..b63ec64970 100644 --- a/tensorflow/contrib/compiler/BUILD +++ b/tensorflow/contrib/compiler/BUILD @@ -14,6 +14,10 @@ py_library( "jit.py", ], srcs_version = "PY2AND3", + deps = [ + "//tensorflow/core:protos_all_py", + "//tensorflow/python:framework_for_generated_wrappers", + ], ) filegroup( diff --git a/tensorflow/contrib/compiler/jit.py b/tensorflow/contrib/compiler/jit.py index 5c84159fcf..f13c1af885 100644 --- a/tensorflow/contrib/compiler/jit.py +++ b/tensorflow/contrib/compiler/jit.py @@ -20,7 +20,8 @@ from __future__ import print_function import contextlib -import tensorflow as tf +from tensorflow.core.framework import attr_value_pb2 +from tensorflow.python.framework import ops @contextlib.contextmanager @@ -43,8 +44,8 @@ def experimental_jit_scope(compile_ops=True): The current scope, enabling or disabling compilation. """ - attrs = {"_XlaCompile": tf.AttrValue(b=compile_ops)} + attrs = {"_XlaCompile": attr_value_pb2.AttrValue(b=compile_ops)} # pylint: disable=protected-access - with tf.get_default_graph()._attr_scope(attrs): + with ops.get_default_graph()._attr_scope(attrs): yield # pylint: enable=protected-access |