diff options
author | Skye Wanderman-Milne <skyewm@google.com> | 2017-09-13 10:49:45 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-09-13 10:54:47 -0700 |
commit | 92362d0f0510d5bb1afa3c9cfd007cbf9cdf6d2a (patch) | |
tree | 6440266f60c78450188586892fc6bab7fa67d859 /tensorflow/cc/BUILD | |
parent | a4f6e7c1afd130d97759b99ba88e69138c59107c (diff) |
Add WhileContext class and add plumbing for creating them.
This change introduces WhileContext, which stores information about a
while loop and will be used in future changes to generate while loop
gradient graphs. Exit nodes in a while loop now have a pointer to
their associated WhileContext. This will be used to retrieve the
context for a given loop.
This change adds an optional parameter to BuildWhileLoop() to create a
WhileContext for the while loop (currently this is always true, but
gradients will generate while loops without associated contexts). This
change also adds a as-yet-unused option to BuildWhileLoop() to return
the predicate output.
PiperOrigin-RevId: 168562303
Diffstat (limited to 'tensorflow/cc/BUILD')
-rw-r--r-- | tensorflow/cc/BUILD | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/cc/BUILD b/tensorflow/cc/BUILD index 028de60880..0d2c9f2d19 100644 --- a/tensorflow/cc/BUILD +++ b/tensorflow/cc/BUILD @@ -258,6 +258,7 @@ tf_cc_test( ":client_session", ":testutil", ":while_loop", + "//tensorflow/core:core_cpu", "//tensorflow/core:test", "//tensorflow/core:test_main", "//tensorflow/core:testlib", |