diff options
Diffstat (limited to 'tensorflow/python/eager/tape.py')
-rw-r--r-- | tensorflow/python/eager/tape.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/tensorflow/python/eager/tape.py b/tensorflow/python/eager/tape.py index a06f5e1a67..afbad183b0 100644 --- a/tensorflow/python/eager/tape.py +++ b/tensorflow/python/eager/tape.py @@ -23,7 +23,6 @@ import contextlib import threading from tensorflow.python import pywrap_tensorflow -from tensorflow.python.util import compat def tid(tensor): @@ -87,9 +86,9 @@ class Tape(object): """Records an operation in the tape.""" pywrap_tensorflow.TFE_Py_TapeRecordOperation( self._tape, - compat.as_bytes(op_type), + op_type, output_tensors, - [x._id for x in input_tensors], # pylint: disable=protected-access + input_tensors, backward_function) def _delete_tensor_id(self, i): |