From 7436b368754f66be9f2803ddee2c9c95ac340357 Mon Sep 17 00:00:00 2001 From: wangyang59 Date: Thu, 19 Apr 2018 16:44:17 -0700 Subject: [PATCH] make bilinear_op registration up-to-date --- paddle/fluid/operators/bilinear_interp_op.cc | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/operators/bilinear_interp_op.cc b/paddle/fluid/operators/bilinear_interp_op.cc index 047a79765fb..a4c3a2a1321 100644 --- a/paddle/fluid/operators/bilinear_interp_op.cc +++ b/paddle/fluid/operators/bilinear_interp_op.cc @@ -10,6 +10,7 @@ limitations under the License. */ #include "paddle/fluid/operators/bilinear_interp_op.h" +#include #include "paddle/fluid/framework/op_registry.h" namespace paddle { @@ -81,8 +82,10 @@ class BilinearInterpOpGrad : public framework::OperatorWithKernel { } // namespace paddle namespace ops = paddle::operators; -REGISTER_OP(bilinear_interp, ops::BilinearInterpOp, ops::BilinearInterpOpMaker, - bilinear_interp_grad, ops::BilinearInterpOpGrad); +REGISTER_OPERATOR(bilinear_interp, ops::BilinearInterpOp, + ops::BilinearInterpOpMaker, + paddle::framework::DefaultGradOpDescMaker); +REGISTER_OPERATOR(bilinear_interp_grad, ops::BilinearInterpOpGrad); REGISTER_OP_CPU_KERNEL(bilinear_interp, ops::BilinearInterpKernel); REGISTER_OP_CPU_KERNEL(bilinear_interp_grad, ops::BilinearInterpGradKernel); -- GitLab