diff options
author | Tom Hennigan <tomhennigan@google.com> | 2018-08-27 03:58:42 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-27 04:02:25 -0700 |
commit | e2c6ec9e86dd86e0dd56e0f11302a5bf5d9ed440 (patch) | |
tree | 2b22f1aac1a91cb3b8f37341836d4ebb6c26ec42 /tensorflow/python/BUILD | |
parent | 8359c1d982ba6efafae4a1855f2e3c05c97cca16 (diff) |
Allow `tf.enable_eager_execution()` to be run multiple times.
Subsequent calls after a successful call to enable eager execution are a no op.
This is mostly to support colab where I might accidentally re-execute this and
have to clear a large stack trace / split my import cell.
PiperOrigin-RevId: 210344960
Diffstat (limited to 'tensorflow/python/BUILD')
-rw-r--r-- | tensorflow/python/BUILD | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index 37af3d350e..f62a927925 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -1348,6 +1348,19 @@ py_test( ) py_test( + name = "framework_ops_enable_eager_test", + size = "small", + srcs = ["framework/ops_enable_eager_test.py"], + main = "framework/ops_enable_eager_test.py", + srcs_version = "PY2AND3", + deps = [ + ":framework", + ":platform_test", + "//tensorflow/python/eager:context", + ], +) + +py_test( name = "framework_tensor_shape_test", size = "small", srcs = ["framework/tensor_shape_test.py"], |