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