def repeat_mapper(node: torch.fx.Node, _: nn.Module) -> Optional[torch.fx.Node]:
    """
    Map repeat to tile.
    """
    with node.graph.inserting_before(node):
        inputs = node.kwargs["input"]
        dims = node.kwargs["sizes"]
        # Skip repeat mapping when the list of dims is not all ints (ie. contains
        # some calculated value). torch.tile cannot support cases where dims
        # are Proxy nodes
        if (
            isinstance(dims, (list, tuple))
            and len(dims) > 0
            and not all(isinstance(x, int) for x in dims)
        ):
            logger.info(
                "Not mapping repeat to an acc op. We can't handle variable dims."
            )
            return
        new_node = node.graph.create_node(
            "call_function",
            tile,
            kwargs={"input": inputs, "dims": dims},
            name=f"{node.name}_repeat_map",
        )
        new_node.meta = node.meta.copy()
        return new_node
Map repeat to tile.
fx2ait/fx2ait/acc_tracer/acc_ops.py