-
Notifications
You must be signed in to change notification settings - Fork 3.6k
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
Fix shapeinference function #2296
Fix shapeinference function #2296
Conversation
Could you please add a test into this file? |
Done. |
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.
Thanks!
@ebarsoum, how does this PR look to you? |
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.
LGTM.
Aside - this is outside the scope of this change, but it may be useful to consider updating the cumsum
op to allow specification of the output type from the user. For this kind of aggregation op it is quite possible that we run into overflow issues. Allowing the specification for output type allows the converter to choose a larger type for internal aggregation the cumulative sum and output.
* Fix shapeinference function * Added shapeinference test for cumsum * update inference test * fix test * minor fix -- shape (1) should be (1,) * Add whitespace after comma to fix flake warning
Updated the shape inference function for CumSum.