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

    Finds module corresponding to the node.
    
coremltools/optimize/torch/quantization/_utils.py