onnxruntime/orttraining
Tang, Cheng ae043e3963
Support ort device tensor in ortmodule's inference (#11112)
* support ort device tensor in ort module inference

* fallback aten equal to cpu; add ortmodule inference test case

* fix python format

Co-authored-by: Cheng Tang <chenta@microsoft.com@orttrainingdev9.d32nl1ml4oruzj4qz3bqlggovf.px.internal.cloudapp.net>
2022-05-03 14:28:30 -07:00
..
orttraining Support ort device tensor in ortmodule's inference (#11112) 2022-05-03 14:28:30 -07:00
pytorch_frontend_examples Set black's target version (#11370) 2022-04-27 14:52:19 -07:00
tools Format all python files under onnxruntime with black and isort (#11324) 2022-04-26 09:35:16 -07:00