-
Notifications
You must be signed in to change notification settings - Fork 99
feat(atenlib): aten_expand supports -1 dim #483
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
feat(atenlib): aten_expand supports -1 dim #483
Conversation
Codecov Report
@@ Coverage Diff @@
## main #483 +/- ##
=======================================
Coverage 72.45% 72.45%
=======================================
Files 109 109
Lines 10601 10602 +1
Branches 1093 1093
=======================================
+ Hits 7681 7682 +1
Misses 2613 2613
Partials 307 307
Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here. |
|
Are there tests we can enable for this? |
Thinking about parametrizing fx-export into runtime test. |
|
Was there an OpInfo covering this case? |
|
|
||
| size = op.Cast(size, to=INT64.dtype) | ||
| # To support -1 dim. | ||
| size = op.Abs(size) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I understand -1 becomes 1. Was that intended or was I missing something?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Only -1 becomes 1. Do you think comment is redundant?
No description provided.