onnxruntime/onnxruntime
Vincent Wang b0699d901c
Support Graph Input and Initializer for GatherToSplit Fusion (#18412)
Support graph input and initializer for GatherToSplit fusion. Previously
the fusion requires Gather nodes consume some other node which cannot be
graph input or initializer.

This helps some model training with such case so that we will not have
GatherGrad in the final graph. GatherGrad is super inefficient in kernel
implementation.
2023-11-15 13:46:38 +08:00
..
contrib_ops onboard MoE (#18279) 2023-11-14 16:48:51 -08:00
core Support Graph Input and Initializer for GatherToSplit Fusion (#18412) 2023-11-15 13:46:38 +08:00
python SDXL demo: consistent opt shape and seed (#18445) 2023-11-14 20:24:32 -08:00
test Support Graph Input and Initializer for GatherToSplit Fusion (#18412) 2023-11-15 13:46:38 +08:00
tool/etw
wasm [js/web/training] Add CreateTrainingSession (#17891) 2023-10-26 09:22:10 -07:00
__init__.py Python API to check whether collective ops are available or not (#17730) 2023-09-29 14:11:05 -07:00
ReformatSource.ps1
ReformatSourcePython.bat
VSCodeCoverage.runsettings