From 7bbfe2aaf8b426d33b181ee07069f44893647f7d Mon Sep 17 00:00:00 2001 From: Peilin Wang Date: Mon, 16 Nov 2020 15:41:34 -0500 Subject: [PATCH] changed divnonan back to PrimitiveWithInfer --- mindspore/ops/operations/math_ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ops/operations/math_ops.py b/mindspore/ops/operations/math_ops.py index 9e5d7b9520..27fd2f4a45 100644 --- 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.