diff options
author | 2018-06-05 13:16:57 -0700 | |
---|---|---|
committer | 2018-06-05 13:19:55 -0700 | |
commit | 1bac6186e19353d9881584ce8ec51bf35d627842 (patch) | |
tree | a784202375b10299c8e86f6e5f3d811e392c71a1 /tensorflow/contrib/BUILD | |
parent | c681be04ec15cdfc225bc61132420781bf23d298 (diff) |
Introduce tf.contrib.control_flow.new_cond.
new_cond is a new implementation of tf.cond. Instead of emitting
control flow ops (i.e. Switch and Merge nodes), new_cond emits a
single If op, which represents the conditional branches as TF
functions.
With this change, users can use new_cond and take its gradient.
The idea is for new_cond to eventually replace tf.cond. There are
several functional and performance gaps that must be addressed first,
including:
* Gradients won't work on imported graphs
* Misc. limitations of TF functions (lack of collections, device scopes, etc.)
PiperOrigin-RevId: 199346735
Diffstat (limited to 'tensorflow/contrib/BUILD')
-rw-r--r-- | tensorflow/contrib/BUILD | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/contrib/BUILD b/tensorflow/contrib/BUILD index 0f9c80404a..50b1ae5cc3 100644 --- a/tensorflow/contrib/BUILD +++ b/tensorflow/contrib/BUILD @@ -31,13 +31,15 @@ py_library( "//tensorflow/contrib/cluster_resolver:cluster_resolver_py", "//tensorflow/contrib/coder:coder_py", "//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", "//tensorflow/contrib/data", - "//tensorflow/contrib/distribute:distribute", "//tensorflow/contrib/deprecated:deprecated_py", + "//tensorflow/contrib/distribute:distribute", "//tensorflow/contrib/distributions:distributions_py", "//tensorflow/contrib/eager/python:tfe", "//tensorflow/contrib/estimator:estimator_py", @@ -83,7 +85,6 @@ py_library( "//tensorflow/contrib/proto", "//tensorflow/contrib/quantization:quantization_py", "//tensorflow/contrib/quantize:quantize_graph", - "//tensorflow/contrib/autograph", "//tensorflow/contrib/receptive_field:receptive_field_py", "//tensorflow/contrib/recurrent:recurrent_py", "//tensorflow/contrib/reduce_slice_ops:reduce_slice_ops_py", |