diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-09-30 23:39:55 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-09-30 23:47:56 -0700 |
| commit | ff18944249f723cf6e2825a3165f1efbb64c4880 (patch) | |
| tree | b2b16931d228162042eb5cf0da614dc4fba83ead /tensorflow/python/eager/tape.py | |
| parent | da8349412fe03c9f55307c7f2674f072073d1b40 (diff) | |
Move EagerTensor from python to C.
PiperOrigin-RevId: 170617321
Diffstat (limited to 'tensorflow/python/eager/tape.py')
| -rw-r--r-- | tensorflow/python/eager/tape.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/python/eager/tape.py b/tensorflow/python/eager/tape.py index e4fdaa111a..84814d48fd 100644 --- a/tensorflow/python/eager/tape.py +++ b/tensorflow/python/eager/tape.py @@ -135,9 +135,9 @@ class Tape(object): # adding an explicit stack if this ever gets out of hand self._delete_tensor_id(tensor_id) - def delete_trace(self, tensor): + def delete_trace(self, tensor_id): """Deletes any trace we have for this tensor.""" - self._delete_tensor_id(tid(tensor)) + self._delete_tensor_id(tensor_id) def export(self): """Exports the internal state of this tape. @@ -237,10 +237,10 @@ def record_operation(op_type, output_tensors, input_tensors, side_outputs, backward_function) -def delete_trace(tensor): +def delete_trace(tensor_id): """Deletes traces for this Tensor from all tapes in the stack.""" for t in _tape_stack.stack: - t.delete_trace(tensor) + t.delete_trace(tensor_id) def top_tape_watched_tensors(): |