mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-23 19:32:23 +00:00
66254eb25a
Add support of GPT2 model optimization: * Match subgraph of Gelu Approximation (using Tanh). * Fuse LayerNormalization if SkipLayerNormalization is not ready. * Output model even if embedding layer is not fused. * Improve Reshape Fusion to improve coverage. * Refine constant input checking, and output fused op counter. Update script according to latest op improvements: * Fusion of Add Bias and Gelu. * Fuse SkipLayerNormalization and Add Bias. Other: * Add ReduceSum for mask as intermediate step. * Refactor verbose setting. |
||
|---|---|---|
| .. | ||
| backend | ||
| datasets | ||
| tools | ||
| __init__.py | ||
| _ld_preload.py | ||
| _pybind_state.py | ||
| exported_symbols.lst | ||
| onnxruntime_pybind_exceptions.h | ||
| onnxruntime_pybind_mlvalue.cc | ||
| onnxruntime_pybind_mlvalue.h | ||
| onnxruntime_pybind_state.cc | ||
| onnxruntime_validation.py | ||
| pybind.def | ||
| session.py | ||
| version_script.lds | ||