| 77 | return placeholder |
| 78 | |
| 79 | def output( |
| 80 | self, |
| 81 | results: list[ProxyValue], |
| 82 | output_kinds: Optional[list[OutputKind]] = None, |
| 83 | output_targets: Optional[list[str | None]] = None, |
| 84 | ) -> ProxyValue: |
| 85 | if output_kinds is None: |
| 86 | output_kinds = [OutputKind.USER_OUTPUT] * len(results) |
| 87 | if output_targets is None: |
| 88 | output_targets = [None] * len(results) |
| 89 | for result, out_kind, target in zip(results, output_kinds, output_targets): |
| 90 | self.output_specs.append( |
| 91 | OutputSpec(out_kind, TensorArgument(result.node.name), target=target) |
| 92 | ) |
| 93 | return super().output(results) |
| 94 | |
| 95 | def get_verifiers(self) -> Optional[list[Verifier]]: |
| 96 | if self.mode == IrMode.ATEN: |