diff options
| author |  Alexandre Passos <apassos@google.com> | 2017-11-10 15:29:35 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-10 17:02:59 -0800 |
| commit | 32ed29515c23bb9ff74d4343693d581754192922 (patch) | |
| tree | ca213f9e44e55f77a768e0bfb154d7fc9cd594c6 /tensorflow/python/eager/tape.py | |
| parent | ce239783908dc2db0eed472344fe8d50ce9a6c9c (diff) | |
Improvement to benchmark.
PiperOrigin-RevId: 175346269
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): |