changed divnonan back to PrimitiveWithInfer

4 years ago · 7bbfe2aaf8
parent 52eb2d3401
commit 7bbfe2aaf8
1 changed files with 1 additions and 1 deletions
--- a/mindspore/ops/operations/math_ops.py
+++ b/mindspore/ops/operations/math_ops.py
@ -1829,7 +1829,7 @@ class Div(_MathBinaryOp):
        return None


-class DivNoNan(PrimitiveWithCheck):
+class DivNoNan(_MathBinaryOp):
    """
    Computes a safe divide which returns 0 if the y is zero.