onnxruntime/orttraining
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
..
orttraining SoftmaxCrossEntropyLossInternalGrad and Sum Fusion (#12746) 2022-09-14 14:45:51 +08:00
pytorch_frontend_examples Set black's target version (#11370) 2022-04-27 14:52:19 -07:00
tools Remove unused orttraining amd dockerfiles and scripts (#12707) 2022-09-02 18:43:21 -07:00