onnxruntime/tools
Tang, Cheng 0adeb86bfd
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>
2021-12-08 13:04:29 -08:00
..
ci_build Fix ortmodule for the pytorch model with ort device (#9927) 2021-12-08 13:04:29 -08:00
doc Add graphviz into Dockerfile images for Python API documentation (#7819) 2021-06-02 16:12:54 -07:00
nuget Openvino ep nuget (#9909) 2021-12-07 07:48:31 -08:00
perf_util
python Update mobile prebuilt package ops to add support for opset 14 and 15 (#9717) 2021-11-18 10:44:39 +10:00