Skip to content

repeat_interleave is missing support for output_size kwarg #2687

New issue
New issue
Open
#2691
Open
repeat_interleave is missing support for output_size kwarg#2687
#2691
Labels
contribution welcomeWe welcome code contributions for thisgood first issueGood for newcomersmodule: torchlibRelated to the torch/aten function lib in development
@yuanyao-nv

Description

@yuanyao-nv
yuanyao-nv
opened on Nov 7, 2025

When converting a graph with repeat_interleave, I'm hitting the following error

<class 'TypeError'>: aten_repeat_interleave_self_int() got an unexpected keyword argument 'output_size'

Looks like onnxscript is missing support for the argument?

onnxscript/onnxscript/function_libs/torch_lib/ops/core.py

Lines 7512 to 7515 in 70e751a

def aten_repeat_interleave_self_int(
self: TensorType, repeats: int, dim: Optional[int] = None
) -> TensorType:
"""repeat_interleave.self_int(Tensor self, SymInt repeats, int? dim=None, *, SymInt? output_size=None) -> Tensor

Metadata

Metadata

Assignees

No one assigned

    Labels

    contribution welcomeWe welcome code contributions for thisgood first issueGood for newcomersmodule: torchlibRelated to the torch/aten function lib in development

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions