class_docstr: Adapted from https://github.com/pytorch/pytorch/blob/master/torch/fx/passes/split_module.py
code: def __init__(self, name: str):
      self.name: str = name
      self.node_names: List[str] = []
      self.inputs: Dict[str, None] = {}
      self.outputs: Dict[str, None] = {}
      self.partitions_dependent_on: Dict[str, None] = {}
      self.partition_dependents: Dict[str, None] = {}
      self.graph: torch.fx.graph.Graph = torch.fx.graph.Graph()
      self.environment: Dict[
          torch.fx.node.Node, torch.fx.node.Node
      ] = {}
      self.targets: Dict[str, Any] = {}
  
file_path: colossalai/colossalai/fx/passes/split_module.py
imports: import inspect
  from typing import Any, Callable, Dict, List, Optional
  import torch
  from packaging import version
  from torch.fx._compatibility import compatibility
  from torch.fx.graph_module import GraphModule
label_desc: Use image node_id 1 to create a new Partition object  with example: obj = Partition(name)