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 6b51b73a5f
Fix ONNX where op for scalar inputs (#2218) 
* Fix ONNX where op dim_inference for scalar inputs

* Rewrite ONNX Where codegen to support scalars

* ONNX Where: Add tests for all_scalar inputs

---------

Co-authored-by: Guillaume Lagrange <lagrange.guillaume.1@gmail.com>
 		2024-09-03 11:17:18 -04:00
..
src Fix ONNX where op for scalar inputs (#2218) 2024-09-03 11:17:18 -04: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.