onnxruntime/orttraining
Wil Brady fa7f80c847
Eager mode: Argmax and fixup max and min. (#11861)
* Eager mode ArgMax support.

* Fix basic max and min functionality with minor generator update. Note this does not address all max and min api scope.

* Add addmm test.
2022-06-23 15:55:34 -04:00
..
orttraining Eager mode: Argmax and fixup max and min. (#11861) 2022-06-23 15:55:34 -04:00
pytorch_frontend_examples Set black's target version (#11370) 2022-04-27 14:52:19 -07:00
tools Update rocm ci to ROCm5.1.1 + torch1.10.0 2022-05-20 11:07:21 +08:00