diff options
author | Alexandre Passos <apassos@google.com> | 2018-07-23 15:11:22 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-23 15:32:59 -0700 |
commit | f85d825500357603afb7a02d2c88ad306ee43006 (patch) | |
tree | a1e90f4af08265f8434a80285ebe4a7c6f8b3e0e /tensorflow/core/common_runtime/direct_session.cc | |
parent | 931a3054d2c13c3438fc58978b3463a0bd268aee (diff) |
Allow differentiating tfe.defun functions which contain conds.
PiperOrigin-RevId: 205732423
Diffstat (limited to 'tensorflow/core/common_runtime/direct_session.cc')
-rw-r--r-- | tensorflow/core/common_runtime/direct_session.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/core/common_runtime/direct_session.cc b/tensorflow/core/common_runtime/direct_session.cc index 44291b0b20..d1fd930d25 100644 --- a/tensorflow/core/common_runtime/direct_session.cc +++ b/tensorflow/core/common_runtime/direct_session.cc @@ -717,7 +717,8 @@ Status DirectSession::Run(const RunOptions& run_options, // Receive outputs. if (outputs) { std::vector<Tensor> sorted_outputs; - const Status s = call_frame.ConsumeRetvals(&sorted_outputs); + const Status s = call_frame.ConsumeRetvals( + &sorted_outputs, /* allow_dead_tensors = */ false); if (errors::IsInternal(s)) { return errors::InvalidArgument(s.error_message()); } else if (!s.ok()) { |