Add support onnx ceil & round

[tye-singwa]

Pull Request Template

Checklist

• Confirmed that cargo run-checks command has been executed.
• Made sure the book is up to date with changes in this PR.

Related Issues/PRs

#1714

Changes

Adding ceil & round onnx support

Testing

Added tests

[codecov]

Codecov Report

Attention: Patch coverage is 99.39759% with 1 line in your changes missing coverage. Please review.

Project coverage is 82.15%. Comparing base (c51045c) to head (0b03514).
Report is 9 commits behind head on main.

Files with missing lines	Patch %	Lines
crates/burn-autodiff/src/ops/tensor.rs	50.00%	1 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #3225      +/-   ##
==========================================
+ Coverage   82.14%   82.15%   +0.01%     
==========================================
  Files         966      970       +4     
  Lines      123257   123421     +164     
==========================================
+ Hits       101247   101402     +155     
- Misses      22010    22019       +9     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

[laggui]

Thanks for the addition! 🙂

This could have been implemented simply as a unary node, but I see that the floor node is also implemented as a separate node. LGTM then! We can simplify in a follow-up PR if needed.

[antimora]

@laggui I want to confirm if this autodiff fix is correct.

I have had a pending comment that I forgot to submit

[laggui]

@laggui I want to confirm if this autodiff fix is correct.

I have had a pending comment that I forgot to submit

Yeah the autodiff implementation previously returned the result with floor for the output (though the gradients were fine). Slipped through the cracks it seems.