code: def find_module(model: _torch.nn.Module, node: _fx.Node):
      """
      Finds module corresponding to the node.
      """
      if hasattr(node, "op") and node.op == "call_module":
          return find_target(model, node.target)
      return None
  
imports: import operator
  from collections import defaultdict
  from enum import Enum
  from typing import Dict
  from typing import List
  from typing import Optional
  import torch
  import torch.ao.quantization
  import torch.fx
  from torch.ao.quantization.backend_config import BackendConfig
  from torch.ao.quantization.backend_config import ObservationType
  from coremltools.optimize.torch._utils.version_utils import is_torch_2
label_desc: Use image node_id 2 for calling a global function with example usage: find_module(model, node) and returns: None, find_target