code: def _get_output_names(gm: torch.fx.GraphModule) -> List[str]:
      output_node = next(n for n in gm.graph.nodes if n.op == "output")
      args = pytree.tree_leaves(output_node.args)
      # if isinstance(args, tuple) and len(args) == 1:
      #     args = args[0]
      return [str(arg) for arg in args]
  
imports: import math
  import unittest
  from typing import List, Set
  import operator
  import torch
  from torch.testing._internal.common_utils import run_tests, TestCase
  from torch.testing import FileCheck
  from torch._dynamo.eval_frame import is_dynamo_supported
  from torch.export import export
  from torch._export.pass_base import _ExportPassBaseDeprecatedDoNotUse
  from torch._export.passes.replace_view_ops_with_view_copy_ops_pass import is_view_op, get_view_copy_of_view_op, ReplaceViewOpsWithViewCopyOpsPass
  from torch._export.passes.functionalize_side_effectful_ops_pass import _FunctionalizeSideEffectfulOpsPass
  from functorch.experimental.control_flow import cond
  from torch.fx.passes.operator_support import OperatorSupport
  from torch.fx.passes.infra.partitioner import Partition
  from torch.utils import _pytree
label_desc: Use image node_id 3 for calling a global function with example usage: _get_output_names(gm) and returns: unknown