onnxruntime
ae043e39 - Support ort device tensor in ortmodule's inference (#11112)

Commit
3 years ago
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>
Author
Parents
Loading