diff options
| author |  Akshay Modi <nareshmodi@google.com> | 2018-02-21 14:42:11 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-21 14:48:12 -0800 |
| commit | 32e39947c80ad410042a5aea266f197e9ecd289d (patch) | |
| tree | 767b0434f701274d98a689d012eb6bb45363e22e /tensorflow/python/pywrap_tfe.i | |
| parent | 24edd50d23416b2c15a4e3509ce079e558500894 (diff) | |
Record gradient in C
PiperOrigin-RevId: 186522240
Diffstat (limited to 'tensorflow/python/pywrap_tfe.i')
| -rw-r--r-- | tensorflow/python/pywrap_tfe.i | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/python/pywrap_tfe.i b/tensorflow/python/pywrap_tfe.i index 50f481d29e..7ab0db5268 100644 --- a/tensorflow/python/pywrap_tfe.i +++ b/tensorflow/python/pywrap_tfe.i @@ -29,9 +29,11 @@ limitations under the License. %rename("%s") TFE_OpNameGetAttrType; %rename("%s") TFE_Py_InitEagerTensor; %rename("%s") TFE_Py_RegisterExceptionClass; +%rename("%s") TFE_Py_RegisterBackwardFunctionGetter; %rename("%s") TFE_Py_RegisterFallbackExceptionClass; %rename("%s") TFE_Py_Execute; %rename("%s") TFE_Py_FastPathExecute; +%rename("%s") TFE_Py_RecordGradient; %rename("%s") TFE_Py_UID; %rename("%s") TFE_Py_TapeSetNew; %rename("%s") TFE_Py_TapeSetRemove; |