diff options
| author |  Alexandre Passos <apassos@google.com> | 2017-10-13 15:03:45 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-10-13 15:07:38 -0700 |
| commit | 0bbdeaf45e07e1f5fb5e15961104e348e3ad8777 (patch) | |
| tree | a8dd281fff7ad3731d40df32c135f0bdf1ba153b /tensorflow/python/pywrap_tfe.i | |
| parent | ba3b33884b5b45096930b3849a2a0c4177f8a42b (diff) | |
Ports the eager gradient tape to C.
The tape stack is still in python as is the backprop code.
PiperOrigin-RevId: 172151189
Diffstat (limited to 'tensorflow/python/pywrap_tfe.i')
| -rw-r--r-- | tensorflow/python/pywrap_tfe.i | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/python/pywrap_tfe.i b/tensorflow/python/pywrap_tfe.i index d5b7294c82..5c624a9c12 100644 --- a/tensorflow/python/pywrap_tfe.i +++ b/tensorflow/python/pywrap_tfe.i @@ -24,6 +24,12 @@ limitations under the License. %rename("%s") TFE_Py_RegisterExceptionClass; %rename("%s") TFE_Py_Execute; %rename("%s") TFE_Py_UID; +%rename("%s") TFE_Py_NewTape; +%rename("%s") TFE_Py_TapeShouldRecord; +%rename("%s") TFE_Py_TapeWatch; +%rename("%s") TFE_Py_TapeDeleteTrace; +%rename("%s") TFE_Py_TapeRecordOperation; +%rename("%s") TFE_Py_TapeExport; %{ |