code: def _forward_pre_hook(
      module_stats: ModuleStats, module: nn.Module, input
  ):
      assert not module_stats.done
      module_stats.start_time = time.time()
  
imports: import time
  from collections import OrderedDict
  from typing import Dict, Sequence
  import functools
  import itertools
  import numpy
  import torch
  import torch.nn
  from .compute_madd import compute_madd
  from .compute_flops import compute_flops
  from .compute_memory import compute_memory
  from .stat_tree import StatTree, StatNode
  from .reporter import report_format
label_desc: Use image node_id 6 for calling a global function with example usage: _forward_pre_hook(module_stats, module, input) without return types