code: def build_graph(model, dummy_input, verbose=False):
      g = TorchProtoGraph(model, dummy_input, verbose)
      return g.graph_def, g.stepstats
  
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 2 for calling a global function with example usage: build_graph(model, dummy_input, verbose) and returns: g,  g