diff --git a/paddle/operators/rowwise_add_op.cc b/paddle/operators/rowwise_add_op.cc index c192da04da717cc41d0451bb12609c512b3d9a8e..a012ab0be06449b6e5340a4c4af520d561db5301 100644 --- a/paddle/operators/rowwise_add_op.cc +++ b/paddle/operators/rowwise_add_op.cc @@ -35,7 +35,7 @@ class RowwiseAddOp : public framework::OperatorWithKernel { class RowwiseAddOpMaker : public framework::OpProtoAndCheckerMaker { public: - RowWiseAddOpMaker(framework::OpProto *proto, + RowwiseAddOpMaker(framework::OpProto *proto, framework::OpAttrChecker *op_checker) : OpProtoAndCheckerMaker(proto, op_checker) { AddInput("X", "The left input of row-wise add op, must be matrix"); @@ -48,9 +48,9 @@ for i in xrange(X.shape[0]): )DOC"); } }; -class RowwiseAddGradOp : public OperatorWithKernel { +class RowwiseAddGradOp : public framework::OperatorWithKernel { protected: - void InferShape(const InferShapeContext &ctx) const override { + void InferShape(const framework::InferShapeContext &ctx) const override { PADDLE_ENFORCE(ctx.InputSize() == 4UL, "RowwiseAddGrad inputs is I, O, OG, size must be 4"); PADDLE_ENFORCE(ctx.OutputSize() == 2, diff --git a/paddle/operators/rowwise_add_op.h b/paddle/operators/rowwise_add_op.h index ad43e753e460e445abd42fb782f9abb9c4853be3..27d7a33e8ad2e87e05d6ec4caf6e484bc7ede054 100644 --- a/paddle/operators/rowwise_add_op.h +++ b/paddle/operators/rowwise_add_op.h @@ -28,7 +28,7 @@ template ; template -class RowwiseAddKernel : public OpKernel { +class RowwiseAddKernel : public framework::OpKernel { public: void Compute(const framework::ExecutionContext& context) const override { auto out = context.Output(0);