saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/orttraining/orttraining
History
Vincent Wang da07c83948
SoftmaxCrossEntropyLossInternalGrad and Sum Fusion (#12746) 
* fuse scegrad and sum

* add yield output shapes to value_info

* resolve comments

* fix merge main
2022-09-14 14:45:51 +08:00
..
core SoftmaxCrossEntropyLossInternalGrad and Sum Fusion (#12746) 2022-09-14 14:45:51 +08:00
eager Pinning manual seed (#12714) 2022-08-25 10:09:02 -07:00
lazy_tensor Make ORT callable from various Pytorch compilers (LazyTensor, TorchDynamo, etc) (#10460) 2022-08-22 09:40:40 -07:00
models Make memory profiler work with multiple session runs. (#12317) 2022-07-29 18:36:31 +08:00
python Add support for ORTModule Torch cpp CUDA extension build within docker (#12868) 2022-09-08 15:30:44 -04:00
test SoftmaxCrossEntropyLossInternalGrad and Sum Fusion (#12746) 2022-09-14 14:45:51 +08:00
training_api Csharp bindings for on-device training APIs (#12404) 2022-09-02 13:13:48 -07:00
training_ops SoftmaxCrossEntropyLossInternalGrad and Sum Fusion (#12746) 2022-09-14 14:45:51 +08:00