Commit 7b04774
authored
[torchlib] Modify aten_unbind to use None for split_sizes (#2536)
According to
https://onnx.ai/onnx/operators/onnx__SplitToSequence.html#summary, `If
the argument split is not specified, a default scalar value of 1 is used
as the value of split`, and this is the only case when `keepdims` can be
set to `0`.
Fixes #25331 parent 8974f5e commit 7b04774
1 file changed
+2
-3
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
8718 | 8718 | | |
8719 | 8719 | | |
8720 | 8720 | | |
8721 | | - | |
| 8721 | + | |
8722 | 8722 | | |
8723 | 8723 | | |
8724 | 8724 | | |
8725 | | - | |
8726 | | - | |
| 8725 | + | |
8727 | 8726 | | |
8728 | 8727 | | |
8729 | 8728 | | |
| |||
0 commit comments