saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-26 19:52:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/orttraining
History
Suffian Khan 76bc0e479c
Enable dense sequence optimized version of Pytorch exported BERT-L on AMD GPU (#6504) 
* Permit dense seq optimization on BERT-L pytorch export by enabling ReduceSumTraining, Equal, and NonZero on AMD

* enable Equal tests

* enable fast_matrix_reduction test case
2021-01-29 13:12:34 -08:00
..
orttraining Enable dense sequence optimized version of Pytorch exported BERT-L on AMD GPU (#6504) 2021-01-29 13:12:34 -08:00
pytorch_frontend_examples Enable running the mnist_training sample without cuda (#6085) 2020-12-15 17:06:54 -08:00
tools Dockerfile to build onnxruntime with ROCm 4.0 2020-12-22 10:21:12 -08:00