Skip to content

Commit 7436b36

Browse files
author
wangyang59
committed
make bilinear_op registration up-to-date
1 parent 4a3c99f commit 7436b36

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

paddle/fluid/operators/bilinear_interp_op.cc

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
limitations under the License. */
1111

1212
#include "paddle/fluid/operators/bilinear_interp_op.h"
13+
#include <vector>
1314
#include "paddle/fluid/framework/op_registry.h"
1415

1516
namespace paddle {
@@ -81,8 +82,10 @@ class BilinearInterpOpGrad : public framework::OperatorWithKernel {
8182
} // namespace paddle
8283

8384
namespace ops = paddle::operators;
84-
REGISTER_OP(bilinear_interp, ops::BilinearInterpOp, ops::BilinearInterpOpMaker,
85-
bilinear_interp_grad, ops::BilinearInterpOpGrad);
85+
REGISTER_OPERATOR(bilinear_interp, ops::BilinearInterpOp,
86+
ops::BilinearInterpOpMaker,
87+
paddle::framework::DefaultGradOpDescMaker<true>);
88+
REGISTER_OPERATOR(bilinear_interp_grad, ops::BilinearInterpOpGrad);
8689
REGISTER_OP_CPU_KERNEL(bilinear_interp, ops::BilinearInterpKernel<float>);
8790
REGISTER_OP_CPU_KERNEL(bilinear_interp_grad,
8891
ops::BilinearInterpGradKernel<float>);

0 commit comments

Comments
 (0)