optimum
3591786b - Support ONNX Runtime optimizations in exporters.onnx (#807)

Commit
3 years ago
Support ONNX Runtime optimizations in exporters.onnx (#807) * test support * fix decoder merge to work with ORT optimizer * add tests * fix test * fix dockerfile * fix * fix doc * fix * properly raise an error at validation * raise error * fix * fix some test * fix bloom validation * fix broken initializer deduplication in case of name collision * Update optimum/onnx/graph_transformations.py Co-authored-by: Michael Benayoun <mickbenayoun@gmail.com> * Update tests/exporters/onnx/test_exporters_onnx_cli.py Co-authored-by: Michael Benayoun <mickbenayoun@gmail.com> * remove print * fix on suggestions * fix remaining tests * Update docs/source/onnxruntime/usage_guides/optimization.mdx Co-authored-by: regisss <15324346+regisss@users.noreply.github.com> * Update optimum/onnx/graph_transformations.py Co-authored-by: Michael Benayoun <mickbenayoun@gmail.com> * Update optimum/exporters/onnx/convert.py Co-authored-by: Michael Benayoun <mickbenayoun@gmail.com> * Update optimum/onnx/graph_transformations.py Co-authored-by: Michael Benayoun <mickbenayoun@gmail.com> --------- Co-authored-by: Michael Benayoun <mickbenayoun@gmail.com> Co-authored-by: regisss <15324346+regisss@users.noreply.github.com>
Author
Parents
Loading