aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/pywrap_tfe.i
diff options
context:
space:
mode:
authorGravatar Alexandre Passos <apassos@google.com>2017-11-13 07:51:58 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-11-13 07:55:57 -0800
commit23e94a4ceb5c0dd2fac8e53d6ceadd2d2b3e9b5e (patch)
tree64e936ab9f7011bcdc8cbe97b110bc5f5f134616 /tensorflow/python/pywrap_tfe.i
parentbd9d8064c8342a59db03301b4afd10238002e344 (diff)
Moves tape.watch_variable to C. Prequel to moving the tape stack to C.
PiperOrigin-RevId: 175531148
Diffstat (limited to 'tensorflow/python/pywrap_tfe.i')
-rw-r--r--tensorflow/python/pywrap_tfe.i2
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/python/pywrap_tfe.i b/tensorflow/python/pywrap_tfe.i
index cbacf458a0..5ca0e57286 100644
--- a/tensorflow/python/pywrap_tfe.i
+++ b/tensorflow/python/pywrap_tfe.i
@@ -30,6 +30,8 @@ limitations under the License.
%rename("%s") TFE_Py_TapeDeleteTrace;
%rename("%s") TFE_Py_TapeRecordOperation;
%rename("%s") TFE_Py_TapeGradient;
+%rename("%s") TFE_Py_TapeWatchVariable;
+%rename("%s") TFE_Py_TapeWatchedVariables;
%rename("%s") TFE_NewContextOptions;
%rename("%s") TFE_ContextOptionsSetConfig;
%rename("%s") TFE_ContextOptionsSetDevicePlacementPolicy;