onnxruntime
0adeb86b - Fix ortmodule for the pytorch model with ort device (#9927)

Commit
4 years ago
Fix ortmodule for the pytorch model with ort device (#9927) * add ortmodule and eager mode test * add ortmodule dependency * convert between aten ort tensor and ortvalue * register the EP to ortmodule using ort device information * remove duplicated test * remove useless dependency * handle half precision type for ortmodule outputs * adjust the tensor conversion python code Co-authored-by: Cheng Tang <chenta@microsoft.com@orttrainingdev9.d32nl1ml4oruzj4qz3bqlggovf.px.internal.cloudapp.net>
Author
Parents
Loading