def expand_as_mapper(node: torch.fx.Node, _: nn.Module) -> torch.fx.Node:
    """
    Maps expand_as(other) to expand(other.size())
    """
    with node.graph.inserting_before(node):
        size_node = node.graph.call_function(
            size, kwargs={"input": node.kwargs["other"]}
        )
        size_node.meta["type"] = torch.Size

        expand_node = node.graph.call_function(
            expand, kwargs={"input": node.kwargs["input"], "sizes": size_node}
        )
        expand_node.meta = node.meta.copy()
        return expand_node
Maps expand_as(other) to expand(other.size())
fx2ait/fx2ait/acc_tracer/acc_ops.py