diff options
Diffstat (limited to 'tensorflow/python/pywrap_tfe.i')
| -rwxr-xr-x | tensorflow/python/pywrap_tfe.i | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/python/pywrap_tfe.i b/tensorflow/python/pywrap_tfe.i index a31861ae40..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; @@ -65,6 +66,7 @@ limitations under the License. %rename("%s") TFE_Py_TensorShapeOnDevice; %rename("%s") TFE_ContextStartStep; %rename("%s") TFE_ContextEndStep; +%rename("%s") TFE_Py_RegisterVSpace; %{ #include "tensorflow/python/eager/pywrap_tfe.h" |