(self, *args, **kwargs)
| 704 | return self._trace.vars |
| 705 | |
| 706 | def _process_inputs(self, *args, **kwargs): |
| 707 | for i, arg in enumerate(args): |
| 708 | assert isinstance( |
| 709 | arg, RawTensor |
| 710 | ), "Only support tensor type args when capture_as_const is enabled" |
| 711 | name_tensor("arg_{}".format(i), arg) |
| 712 | |
| 713 | # TODO: mark kwargs in order |
| 714 | for k, kwarg in kwargs.items(): |
| 715 | if isinstance(kwarg, RawTensor): |
| 716 | name_tensor("kwarg_{}".format(k), kwarg) |
| 717 | |
| 718 | if self._arg_bindings is None: |
| 719 | self._arg_bindings = [ |
| 720 | ("arg_{}".format(i), arg._tuple_shape) for i, arg in enumerate(args) |
| 721 | ] |
| 722 | |
| 723 | if self._kwarg_bindings is None: |
| 724 | self._kwarg_bindings = { |
| 725 | "kwarg_{}".format(k): (k, kwarg._tuple_shape) |
| 726 | for k, kwarg in kwargs.items() |
| 727 | if isinstance(kwarg, RawTensor) |
| 728 | } |
| 729 | |
| 730 | def _process_outputs(self, outputs): |
| 731 | assert ( |
no test coverage detected