code: def get_module_flops(module):
      sum = module.__flops__
      # iterate over immediate children modules
      for child in module.children():
          sum += get_module_flops(child)
      return sum
  
imports: import time
  import torch
  import torch.nn
  import torch.nn.functional
  from functools import partial
  from typing import List, Optional
  from collections import OrderedDict
  import numpy
label_desc: Use image node_id 42 for calling a global function with example usage: get_module_flops(module) and returns: sum