code: def __init__(self, name: str):
      self.name: str = name
      self.submod_name = f"submod_{name}"
      self.node_names: List[str] = []
      self.inputs: Dict[str, None] = {}
      self.outputs: Dict[str, None] = {}
      self.dependencies: Dict[str, None] = {}
      self.dependents: Dict[str, None] = {}
      self.graph: torch.fx.graph.Graph = torch.fx.graph.Graph()
      self.environment: Dict[Node, Node] = {}
      self.targets: Dict[str, Any] = {}
  
file_path: pytorch/torch/fx/passes/split_module.py
imports: import inspect
  from typing import Any, Callable, Dict, List, Optional, Set, TYPE_CHECKING
  from collections import OrderedDict
  import logging
  import torch
  from torch.fx._compatibility import compatibility
  from torch.fx.graph_module import GraphModule
  from torch.fx.node import Node
label_desc: Use image node_id 1 to create a new Partition object  with example: obj = Partition(name)