Commit Graph

8 Commits

Author SHA1 Message Date
Dilshod Tadjibaev 947bb60fff
Add ConvTranspose2d ONNX OP (#1018)
* [~] burn-import: ConvTranspose initial support added.

* Complete conv transpose2d ONNX op

* Fix tests

* Changes per PR feedback

---------

Co-authored-by: Никита Пацакула <pacakula@dml001.kribrum.ru>
2023-11-30 12:22:06 -06:00
Dilshod Tadjibaev fed618b265
Add Cos, Exp, Gelu, Log and Neg unary ONNX ops (#1013)
* Add ONNX Cos OP

* Add ONNX Exp OP

* Add ONNX Gelu OP

* Add ONNX Log OP

* Allow approx_constant clippy rule in generated model code

* Add ONNX Neg OP

* Fix tests with custom imports for unary nodes

* Add scalar tests for Sqrt
2023-11-29 14:57:52 -06:00
David Chavez fda9e5d388
feature(tensor): Add triu and tril ops (#1002) 2023-11-29 09:39:42 -05:00
Edmondo Porcu b86bc58761
Implementing ONNX support for sqrt (#991)
* Implementing ONNX support for sqrt

* Formatting, my bad

* Implementing feedback from pull request

* Fixing codegen

* Fixing tests

* Fixing tests
2023-11-22 16:12:35 -06:00
Zsombor 4fc0c27e31
Implement tensor.recip() function to calculate elementwise reciprocals (#953) 2023-11-15 09:17:32 -05:00
Ariel Cohen cb4c23b39d
add gather (#947) 2023-11-13 08:12:00 -05:00
AuruTus 84e74df3b9
Feat/onnx import erf (#859) 2023-10-13 09:59:44 -04:00
Dilshod Tadjibaev bb71f17409
Update burn-import README (#727) 2023-08-30 08:20:53 -04:00