onnxruntime/orttraining
Sherlock 9174cbe3d5
Optimize CUDA Kernel for 3D and 4D Transpose (#8928)
* Optimize Transpose120 and Transpose102

* Generalize Transpose0123 for more input shapes

* Add Transpose3D test cases

* update rocm kernel
2021-09-13 23:00:53 -07:00
..
orttraining Optimize CUDA Kernel for 3D and 4D Transpose (#8928) 2021-09-13 23:00:53 -07:00
pytorch_frontend_examples Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
tools Add hugging-face models loss curve and performance guards to ROCm CI pipeline. (#8915) 2021-09-01 09:03:10 -07:00