onnxruntime/onnxruntime
Anton Korablin 207c57219a
Add support for full ViT optimization (#15289)
Add support for ViT optimization in optimizer.py
As ViT architecture follows BERT rather closely, we can easily reuse
BERT fusions for ViT. The only difference is that ViT does not have
attention mask, which means there is no Add node in qk paths.
Make the necessary changes in onnx_exporter.py to be able to cover
optimizations with test.
2023-04-04 14:05:24 -07:00
..
contrib_ops fix build bug when enabling DEBUG_GENERATION (#15338) 2023-04-04 09:44:07 -07:00
core Adds int32_t and uint32_t clip kernels (#15306) 2023-04-04 13:44:50 -07:00
python Add support for full ViT optimization (#15289) 2023-04-04 14:05:24 -07:00
test Add support for full ViT optimization (#15289) 2023-04-04 14:05:24 -07:00
tool/etw
wasm [js/web] support flag 'optimizedModelFilePath' in session options (#14355) 2023-02-24 15:50:15 -08:00
__init__.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
ReformatSource.ps1
ReformatSourcePython.bat
VSCodeCoverage.runsettings