dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
burn/crates/onnx-ir
History
Adrian Müller c2b6318fc3
Implement ONNX ConstantOfShape (#1815) 
* Feat: burn-import implement ONNX ConstantOfShape

* Introduce shape type and use in ConstantOfShape and Shape

* Add tests for bool and int tensors for ConstantOfShape

* Fix ONNX test generation

* Undo comment

---------

Co-authored-by: Dilshod Tadjibaev <939125+antimora@users.noreply.github.com>
 		2024-07-08 15:11:59 -05:00
..
src Implement ONNX ConstantOfShape (#1815) 2024-07-08 15:11:59 -05:00
Cargo.toml Separating ONNX parsing from burn-import (#1921) 2024-07-02 15:17:44 -05:00
README.md Separating ONNX parsing from burn-import (#1921) 2024-07-02 15:17:44 -05:00
build.rs Separating ONNX parsing from burn-import (#1921) 2024-07-02 15:17:44 -05:00

README.md

ONNX-IR

A pure rust Onnx Parser. Creates an intermediate representation useful for generating code in any ML/DL framework

For a full list of currently supported operators, please check here

To see how to use this for generating burn graphs, see here.