onnxruntime
6ceee5d1 - IsInf ReduceSum transform (#7188)

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