burn/crates/onnx-ir
johnhuichen 4a3fc9d4a0
Implement ONNX Pad Operator (#2007)
* Implement ONNX pad

* ONNX pad arguments fix

pad now requires 2 or more arguments
if the third argument is not given, it will default to 0

* fixing bug in input len fix

* change panic comment

Change panic comment from needing two inputs. This comes from the fact that the ONNX spec requires two necessary inputs but could have more two more optional argument.

---------

Co-authored-by: JC <you@example.com>
Co-authored-by: mepatrick73 <pameu17@ulaval.ca>
2024-07-23 13:50:20 -04:00
..
src Implement ONNX Pad Operator (#2007) 2024-07-23 13:50:20 -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.