onnxruntime/onnxruntime
Vincent Wang 0c078dfc8b
Some Shape Related Fusions (#19832)
This PR adds below shape related fusions, which is helpful for some
transformer models:
- ShapeInputMerge is to merge all Shape nodes' input NodeArg to a single
one (the 1st one on topo order) if they have the same shape value. This
helps CSE fusion to merge more nodes.
- CSE fusion to support scalar tensor as attribute value. This is mainly
to support ConstantOfShape node.
2024-03-12 10:29:27 +08:00
..
contrib_ops Remove memset for the case no any mask (#19823) 2024-03-07 13:54:16 -08:00
core Some Shape Related Fusions (#19832) 2024-03-12 10:29:27 +08:00
python [EP Perf] Update EP Perf dockerfiles with cuda12/cudnn9 (#19781) 2024-03-08 13:58:22 -08:00
test Some Shape Related Fusions (#19832) 2024-03-12 10:29:27 +08:00
tool/etw
wasm [js/webgpu] Support capture and replay for jsep (#18989) 2024-01-30 18:28:03 -08:00
__init__.py
ReformatSource.ps1
ReformatSourcePython.bat
VSCodeCoverage.runsettings