diff options
author | Skye Wanderman-Milne <skyewm@google.com> | 2018-06-15 14:54:00 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-15 14:57:08 -0700 |
commit | 94b3db68ee2edb568b6b12d3063b72074910f878 (patch) | |
tree | c9ef7668e327580e7063111335e7a2495d4d4e8c /tensorflow/contrib/BUILD | |
parent | 33f8f7e1843c750186c8fbcfbf94f286bb7ca505 (diff) |
Move cond_v2 to core (non-public) and add toggle to use cond_v2 by default.
This change:
* Creates a new global variable, control_flow_ops._ENABLE_COND_V2, to use
cond_v2 by default when calling tf.cond. This variable can also be
controlled via the environment variable TF_ENABLE_COND_V2.
* Moves cond_v2 out of contrib so it's accessible from control_flow_ops.py.
* Lazily "imports" some modules in cond_v2 to avoid circular dependencies.
Note that these lazy "imports" must be imported by the cond_v2 caller (or
recursively by one of the caller's imports) in order for cond_v2 to have
access to them.
* Renames the cond_v2 module to cond_v2_impl, and creates a new cond_v2 module
that imports the cond_v2 method and the necessary extra imports. This is
useful for explicitly calling cond_v2 outside of control_flow_ops.cond.
PiperOrigin-RevId: 200778208
Diffstat (limited to 'tensorflow/contrib/BUILD')
-rw-r--r-- | tensorflow/contrib/BUILD | 1 |
1 files changed, 0 insertions, 1 deletions
diff --git a/tensorflow/contrib/BUILD b/tensorflow/contrib/BUILD index 50b1ae5cc3..7d44a054a8 100644 --- a/tensorflow/contrib/BUILD +++ b/tensorflow/contrib/BUILD @@ -33,7 +33,6 @@ py_library( "//tensorflow/contrib/compiler:compiler_py", "//tensorflow/contrib/autograph", "//tensorflow/contrib/constrained_optimization", - "//tensorflow/contrib/control_flow", "//tensorflow/contrib/copy_graph:copy_graph_py", "//tensorflow/contrib/crf:crf_py", "//tensorflow/contrib/cudnn_rnn:cudnn_rnn_py", |