diff options
| author |  Akshay Modi <nareshmodi@google.com> | 2018-10-01 15:10:19 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-01 15:16:07 -0700 |
| commit | c7237e6070dbf4acd1ade5a40dc676418cbd889b (patch) | |
| tree | 9671709249d76f83b273192e9d729d20d8135910 /tensorflow/python/eager | |
| parent | 9084e999b3caf65833f9651c72bc09eb3094eba5 (diff) | |
Don't generate backward function and delete when its not necessary
PiperOrigin-RevId: 215288224
Diffstat (limited to 'tensorflow/python/eager')
| -rw-r--r-- | tensorflow/python/eager/pywrap_tfe_src.cc | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/python/eager/pywrap_tfe_src.cc b/tensorflow/python/eager/pywrap_tfe_src.cc index 4b9f7f4100..ae1e12f9c3 100644 --- a/tensorflow/python/eager/pywrap_tfe_src.cc +++ b/tensorflow/python/eager/pywrap_tfe_src.cc @@ -1567,9 +1567,8 @@ void TapeSetRecordOperation( } for (TFE_Py_Tape* tape : SafeTapeSet()) { - auto* function = backward_function_getter(); tape->tape->RecordOperation(op_type_str, output_info, input_ids, - input_dtypes, function, + input_dtypes, backward_function_getter, backward_function_killer); } } |