onnxruntime/orttraining
ashbhandare 6ceee5d131
IsInf ReduceSum transform (#7188)
* IsInf ReduceSum transform

* Revert unnecessary changes, add isinf_only and isnan_only attr

* add tests, review comments

* Disable test for non-cuda

* Move IsAllFinite from training to contrib op

* review comments

* Review comment, formatting

* Enable test for ROCm EP
2021-04-13 16:05:21 -07:00
..
orttraining IsInf ReduceSum transform (#7188) 2021-04-13 16:05:21 -07:00
pytorch_frontend_examples Sync ORTModule branch with master and fix tests (#6526) 2021-02-02 08:59:56 -08:00
tools Add BERT-L perf regression test on MI100 and re-enable batch size test (#7240) 2021-04-05 15:51:52 -07:00