def dropout_mapper(node: torch.fx.Node, mod: nn.Module):
    """
    Remove dropout node and directly map its input to output.
    """
    return node.kwargs["input"]
Remove dropout node and directly map its input to output.
fx2ait/fx2ait/acc_tracer/acc_ops.py