def METHOD_NAME(gm, node: torch.fx.Node):
    if node.op == "call_function":
        return node.target.__name__
    elif node.op in ["placeholder", "call_method"]:
        return node.target
    elif node.op == "call_module":
        return gm.get_submodule(node.target).__class__.__name__
    elif node.op == "get_attr":
        return node.target
    elif node.op == "output":
        return "output"
    raise AssertionError(node.op)