-
Notifications
You must be signed in to change notification settings - Fork 645
Floor Node #2792
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
Floor Node #2792
Conversation
Codecov ReportAll modified and coverable lines are covered by tests ✅
Additional details and impacted files@@ Coverage Diff @@
## main #2792 +/- ##
==========================================
+ Coverage 83.67% 83.68% +0.01%
==========================================
Files 832 833 +1
Lines 109524 109605 +81
==========================================
+ Hits 91643 91724 +81
Misses 17881 17881 ☔ View full report in Codecov by Sentry. |
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 for the addition 🙏
Some minor comments, otherwise will be good to go!
We should also mark the op as supported here: https://github.com/tracel-ai/burn/blob/main/crates/burn-import/SUPPORTED-ONNX-OPS.md
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.
Thank you for taking up this task.
A couple of small changes needed:
- Update the supported ops doc for
floor
op:crates/burn-import/SUPPORTED-ONNX-OPS.md
- Remove unrelated changes (probably accidentally checked in):
crates/burn-import/onnx-tests/tests/relu/relu.onnx
crates/burn-import/onnx-tests/tests/one_hot/one_hot.onnx
crates/burn-import/onnx-tests/tests/one_hot/one_hot.py
Ran the tests this time. 🥲 |
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 for contributing 🙏
Pull Request Template
Checklist
run-checks all
script has been executed.Related Issues/PRs
#1714
#1544
Changes
Implemented Floor Node
Testing
Ran run_checks and checked the code separately too.