def METHOD_NAME(module: torch.fx.GraphModule, input: Any) -> torch.fx.GraphModule:
    for pass_ in passes:
        if isinstance(module, torch.fx.GraphModule):
            ShapeProp(module).propagate(*input)
        module = pass_(module)
    return module