def METHOD_NAME(model, graph):
  module_to_node = {}
  for node in graph.nodes:
    module = mod_util.get_module_by_node(model, node)
    if module:
      module_to_node[id(module)] = node.name
  return module_to_node