code: def build_module_graph(model, dummy_input):
      return TorchModuleGraph(model, dummy_input)
  
imports: import logging
  import queue
  import re
  from collections import defaultdict
  from typing import List, Dict
  import torch
  from torch.utils.tensorboard._pytorch_graph import NodePy, NodePyIO, NodePyOP, GraphPy
label_desc: Use image node_id 1 for calling a global function with example usage: build_module_graph(model, dummy_input) and returns: TorchModuleGraph