未验证 提交 85b2fa4b 编写于 作者: J Jiabin Yang 提交者: GitHub

Fix relu custom vjp (#51838)

* support relue custom vjp

* Fix some typo issue for relu custom vjp
上级 7581ef9e
...@@ -65,8 +65,8 @@ def expect_grad(inputs): ...@@ -65,8 +65,8 @@ def expect_grad(inputs):
return gradients return gradients
class TestCompositeSoftmaxPrimBackward(unittest.TestCase): class TestCompositeReluPrimBackward(unittest.TestCase):
"test composite softmax and prim backward" "test composite relu and prim backward"
def setUp(self): def setUp(self):
core._set_prim_backward_enabled(True) core._set_prim_backward_enabled(True)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册