diff options
| author |  Alexandre Passos <apassos@google.com> | 2018-05-17 08:23:10 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-17 08:25:40 -0700 |
| commit | 9a75743f7a4190c788a33ec7bd4b384e12292cb1 (patch) | |
| tree | 5a82790fb7b08a10df0aefa34d89a191e845d41d /tensorflow/python/pywrap_tfe.i | |
| parent | ec0659b06a07f010123039baa16d120bd0752dda (diff) | |
Methods to stop and reset tf.GradientTape()
PiperOrigin-RevId: 196995160
Diffstat (limited to 'tensorflow/python/pywrap_tfe.i')
| -rw-r--r-- | tensorflow/python/pywrap_tfe.i | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/pywrap_tfe.i b/tensorflow/python/pywrap_tfe.i index 5ee55301df..fde3223e96 100644 --- a/tensorflow/python/pywrap_tfe.i +++ b/tensorflow/python/pywrap_tfe.i @@ -42,6 +42,7 @@ limitations under the License. %rename("%s") TFE_Py_RecordGradient; %rename("%s") TFE_Py_UID; %rename("%s") TFE_Py_TapeSetNew; +%rename("%s") TFE_Py_TapeSetAdd; %rename("%s") TFE_Py_TapeSetRemove; %rename("%s") TFE_Py_TapeSetStopOnThread; %rename("%s") TFE_Py_TapeSetRestartOnThread; |