onnxruntime/orttraining
Tang, Cheng 7660eeef3e
fix ortmodule's output device info when it runs on ort device (#10616)
Co-authored-by: Cheng Tang <chenta@microsoft.com@orttrainingdev9.d32nl1ml4oruzj4qz3bqlggovf.px.internal.cloudapp.net>
2022-02-24 10:22:55 -08:00
..
orttraining fix ortmodule's output device info when it runs on ort device (#10616) 2022-02-24 10:22:55 -08:00
pytorch_frontend_examples
tools Enable OneHot kernel for ROCm EP and add Dockerfile for ROCm 4.3.1 (#9656) 2021-12-07 12:47:00 -08:00