diff options
author | Tom Hennigan <tomhennigan@google.com> | 2018-09-07 02:56:32 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-07 03:00:56 -0700 |
commit | c35e6928dc0b00ca3ca489e5d34f856499cece4a (patch) | |
tree | 799f8dffe4ea6bf096dcd4ecc178bcf34f4ed19f /tensorflow/python/pywrap_tfe.i | |
parent | 5a635e3472e16007830fca533c35b2f63fc4f898 (diff) |
Support not automatically watching (trainable) accessed variables in GradientTape.
For more complex use cases this allows fine grained control over what is tracked
by the tape.
PiperOrigin-RevId: 211948236
Diffstat (limited to 'tensorflow/python/pywrap_tfe.i')
-rwxr-xr-x | tensorflow/python/pywrap_tfe.i | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/python/pywrap_tfe.i b/tensorflow/python/pywrap_tfe.i index 2253edc742..be8f425481 100755 --- a/tensorflow/python/pywrap_tfe.i +++ b/tensorflow/python/pywrap_tfe.i @@ -52,9 +52,10 @@ limitations under the License. %rename("%s") TFE_Py_TapeSetShouldRecord; %rename("%s") TFE_Py_TapeSetDeleteTrace; %rename("%s") TFE_Py_TapeSetRecordOperation; -%rename("%s") TFE_Py_TapeSetWatchVariable; %rename("%s") TFE_Py_TapeGradient; +%rename("%s") TFE_Py_TapeVariableAccessed; %rename("%s") TFE_Py_TapeWatch; +%rename("%s") TFE_Py_TapeWatchVariable; %rename("%s") TFE_Py_TapeWatchedVariables; %rename("%s") TFE_NewContextOptions; %rename("%s") TFE_ContextOptionsSetConfig; |