onnxruntime/orttraining
Vincent Wang 2bed0d4abb
[CUDA] SoftmaxCrossEntropy Kernels Refactor (#12482)
* sce refactor

* refactor

* remove usnecessory memset
2022-08-09 16:48:44 +08:00
..
orttraining [CUDA] SoftmaxCrossEntropy Kernels Refactor (#12482) 2022-08-09 16:48:44 +08:00
pytorch_frontend_examples Set black's target version (#11370) 2022-04-27 14:52:19 -07:00
tools [UPDATE] update AMD CI pipeline to Rocm5.2 with torch1.11 (#12162) 2022-07-14 16:38:16 +08:00