code: def __init__(self, gm: torch.fx.GraphModule) -> None:
      super().__init__()
      self.gm = gm
      self.compiled: Optional[
          Callable[[List[torch.Tensor]], List[torch.Tensor]]
      ] = None
  
file_path: pytorch/torch/distributed/_spmd/partial_lower.py
imports: import logging
  import operator
  from typing import Callable, List, Optional, Set, Tuple
  from functorch import make_fx
  import torch
  from torch._inductor.compile_fx import compile_fx_inner
  from torch._inductor.decomposition import select_decomp_table
label_desc: Use image node_id 1 to create a new _InductorModule object from inherited base classes: torch.nn with example: obj = _InductorModule(gm)