Capture a specific tensor and utilize the provided placeholder. Args: tensor: Tensor to captures. placeholder: Provided placeholder for the tensor.
(self, tensor, placeholder)
| 623 | return self._captures.values() |
| 624 | |
| 625 | def add_capture(self, tensor, placeholder): |
| 626 | """Capture a specific tensor and utilize the provided placeholder. |
| 627 | |
| 628 | Args: |
| 629 | tensor: Tensor to captures. |
| 630 | placeholder: Provided placeholder for the tensor. |
| 631 | """ |
| 632 | self._captures[ops.tensor_id(tensor)] = (tensor, placeholder) |
| 633 | self.inputs.append(placeholder) |
| 634 | |
| 635 | def reset_captures(self, capture_list): |
| 636 | """Set the captures with the provided list of captures & placeholder.""" |
no test coverage detected