Remove the capture and return the generated placeholder.
(self, tensor)
| 639 | self._captures[ops.tensor_id(tensor)] = (tensor, placeholder) |
| 640 | |
| 641 | def pop_capture(self, tensor): |
| 642 | """Remove the capture and return the generated placeholder.""" |
| 643 | capture = self._captures.pop(ops.tensor_id(tensor), None) |
| 644 | if capture is None: |
| 645 | return None |
| 646 | |
| 647 | return capture[1] |
| 648 | |
| 649 | def clear_captures(self): |
| 650 | # TODO(b/115366440): Delete this method when a custom OrderedDict is added. |
no test coverage detected