We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8974f5e commit 7b04774Copy full SHA for 7b04774
onnxscript/function_libs/torch_lib/ops/core.py
@@ -8718,12 +8718,11 @@ def aten_type_as(self: TTensor, other: TTensor2) -> TTensor2:
8718
return op.CastLike(self, other)
8719
8720
8721
-@torch_op("aten::unbind.int")
+@torch_op("aten::unbind.int", trace_only=True)
8722
def aten_unbind(self: TTensor, dim: int = 0) -> Sequence[TTensor]:
8723
"""unbind.int(Tensor(a -> *) self, int dim=0) -> Tensor(a)[]"""
8724
8725
- split_sizes = op.Constant(value_int=1)
8726
- return op.SplitToSequence(self, split_sizes, axis=dim, keepdims=False)
+ return op.SplitToSequence(self, axis=dim, keepdims=False)
8727
8728
8729
@torch_op("aten::unflatten.int", trace_only=True)
0 commit comments