diff options
| author |  Suharsh Sivakumar <suharshs@google.com> | 2017-10-04 00:07:16 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-10-04 00:11:41 -0700 |
| commit | d016cb020583b1ecbc260c1492e347c2731b1c29 (patch) | |
| tree | cffe8e044b948ef3a5c0dde1c752b550d4e193de /tensorflow/cc/BUILD | |
| parent | f9f037c1c489d6a72ef682e3bce01e6f154222e4 (diff) | |
Fix c++ gradients issue where multiple dependent outputs result in incorrect answer.
The issue is that we incorrectly calculate the pending num_expected_backprops for outputs nodes when one output transitively depends on another. this is because we use output nodes as an indicator of when we need to end our traversal. Instead we should only use output nodes that don't transitively get consumed by other output nodes as end indicators for our traversal. This change implements that fix.
Fixes #13190
PiperOrigin-RevId: 170971937
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 3682ebd943..80112f9b44 100644 --- a/tensorflow/cc/BUILD +++ b/tensorflow/cc/BUILD @@ -45,6 +45,7 @@ tf_cc_test( srcs = ["framework/gradients_test.cc"], deps = [ ":cc_ops", + ":client_session", ":grad_op_registry", ":grad_ops", ":gradients", |