MCPcopy Create free account
hub / github.com/pytorch/executorch / output

Method output

backends/test/program_builder.py:79–93  ·  view source on GitHub ↗
(
        self,
        results: list[ProxyValue],
        output_kinds: Optional[list[OutputKind]] = None,
        output_targets: Optional[list[str | None]] = None,
    )

Source from the content-addressed store, hash-verified

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:

Calls 3

OutputSpecClass · 0.85
TensorArgumentClass · 0.85
appendMethod · 0.45