提交 260026fa 编写于 作者: Z zchen0211

prelu modify

上级 58b5b08b
...@@ -16,3 +16,6 @@ ...@@ -16,3 +16,6 @@
REGISTER_OP_GPU_KERNEL( REGISTER_OP_GPU_KERNEL(
prelu, paddle::operators::PreluKernel<paddle::platform::GPUPlace, float>); prelu, paddle::operators::PreluKernel<paddle::platform::GPUPlace, float>);
REGISTER_OP_GPU_KERNEL(
prelu_grad,
paddle::operators::PreluGradKernel<paddle::platform::GPUPlace, float>);
...@@ -3,7 +3,7 @@ import numpy as np ...@@ -3,7 +3,7 @@ import numpy as np
from op_test import OpTest from op_test import OpTest
class ScaleTest(OpTest): class PreluTest(OpTest):
def setUp(self): def setUp(self):
self.op_type = "prelu" self.op_type = "prelu"
self.inputs = {'X': np.random.random((10, 10)).astype("float32")} self.inputs = {'X': np.random.random((10, 10)).astype("float32")}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册