onnxruntime/orttraining
Weixing Zhang 53307a5f2e
improve perf for softmax (#6128)
* improve perf for both gathergrad and softmax

* revert the change in gathergrad and will be done in another PR.

* address comments from code review.
2020-12-21 14:15:54 -08:00
..
orttraining improve perf for softmax (#6128) 2020-12-21 14:15:54 -08:00
pytorch_frontend_examples Enable running the mnist_training sample without cuda (#6085) 2020-12-15 17:06:54 -08:00
tools add dockerfile for ROCm3.10 and update BUILD.md for ROCm EP (#5821) 2020-12-08 23:14:56 -08:00