code: def __init__(
      self, trace_indice: TraceIndice, node_mgr: NodeMgr
  ) -> None:
      self.trace_indice = trace_indice
      self.node_mgr = node_mgr
  
file_path: colossalai/colossalai/autochunk/trace_flow.py
imports: from typing import Dict, List, Tuple
  from torch.fx.node import Node
  from .trace_indice import TraceIndice
  from .utils import NodeMgr, find_chunk_all_input_nodes, find_chunk_compute_input_and_output_nodes, find_tensor_shape_node, flat_list, get_node_name, get_node_shape, is_non_compute_node
label_desc: Use image node_id 1 to create a new TraceFlow object from inherited base classes: object with example: obj = TraceFlow(trace_indice, node_mgr)