Searched refs:module_outputs (Results 1 – 1 of 1) sorted by relevance
567 def module_outputs(self) -> Sequence[torch.fx.Node]: member in _ModuleNode584 module_outputs: dict[torch.fx.Node, None] = {}589 module_outputs[node] = None590 return list(module_outputs.keys())625 ref_submodule_outputs = ir_node.module_outputs()681 ref_submodule_outputs = self.module_outputs()682 new_outputs = [copy_env[ref_output] for ref_output in self.module_outputs()]