!11677 add isfinite grad

From: @jiangzg001
Reviewed-by: @liangchenghui,@c_34
Signed-off-by: @liangchenghui
pull/11677/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 58b8d8cd02

@ -778,6 +778,16 @@ def get_bprop_reduce_mean(self):
return bprop
@bprop_getters.register(P.IsFinite)
def get_bprop_isfinite(self):
"""Grad definition for `IsFinite` operation."""
def bprop(x, out, dout):
return (zeros_like(x),)
return bprop
@bprop_getters.register(P.Equal)
def get_bprop_equal(self):
"""Grad definition for `Equal` operation."""

@ -1401,6 +1401,10 @@ test_case_math_ops = [
'block': P.RealDiv(),
'desc_inputs': [[4, 5], [2, 3, 4, 5]],
'desc_bprop': [[2, 3, 4, 5]]}),
('IsFinite', {
'block': P.IsFinite(),
'desc_inputs': [Tensor(np.random.random((3, 4, 5)).astype(np.float32))],
'desc_bprop': [Tensor(np.random.random((3, 4, 5)).astype(np.bool))]}),
('Div', {
'block': P.Div(),
'desc_inputs': [[4, 5], [2, 3, 4, 5]],

Loading…
Cancel
Save