diff options
| author |  Saurabh Saxena <srbs@google.com> | 2018-09-28 12:46:10 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-28 12:53:29 -0700 |
| commit | 5e66d25666aad9fa76ed8cc0d2b162db76ea0cc8 (patch) | |
| tree | a5b810c506ee8eb61b707e890ae41b71ac4cb8bd /tensorflow/python/BUILD | |
| parent | e00954e8626c74b263b90527e0c020cfd64136b2 (diff) | |
Add flag for enabling while_v2.
Add a single test flag for enabling v2 control flow in tests since we do not plan to support v2 ops with legacy control flow.
We have 2 test decorators now:
@with_control_flow_v2: Enables all tests in a class to run with v2 control flow.
@disable_control_flow_v2: Disables a test function from running in v2. I have removed the skiptests to avoid setup/teardown overheads.
Enable tests in control_flow_ops_py_test that run with control_flow_v2.
PiperOrigin-RevId: 214980108
Diffstat (limited to 'tensorflow/python/BUILD')
| -rw-r--r-- | tensorflow/python/BUILD | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index 91cafea042..9275ad767e 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -2017,6 +2017,7 @@ py_library( ":array_ops", ":cond_v2_impl", ":constant_op", + ":control_flow_ops", ":control_flow_util", ":framework_ops", ":function_def_to_graph", |