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 e8ea9e27c2
Improve ONNX import tensor shape tracking (#2213) 
- Calculate result of broadcasting in dim_inference
- keep Shape info when converting from Argument to TensorType
- Remove a few sources of Dim = 0 Tensors, create Scalars instead
- Clean up dim_inference a bit
 		2024-08-29 14:06:30 -04:00
..
src Improve ONNX import tensor shape tracking (#2213) 2024-08-29 14:06:30 -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.