diff --git a/paddle/operators/adamax_op.cc b/paddle/operators/adamax_op.cc
index c348e0a0b2ba1a76be4ed3c54f3e6f91d1767794..5cf727742c770f8f0d59cd49345c5c055722a56c 100644
--- a/paddle/operators/adamax_op.cc
+++ b/paddle/operators/adamax_op.cc
@@ -22,7 +22,7 @@ class AdamaxOp : public framework::OperatorWithKernel {
   using framework::OperatorWithKernel::OperatorWithKernel;
 
  protected:
-  void InferShape(framework::InferShapeContextBase *ctx) const override {
+  void InferShape(framework::InferShapeContext *ctx) const override {
     PADDLE_ENFORCE(ctx->HasInput("Param"),
                    "Input(Param) of AdamaxOp should not be null.");
     PADDLE_ENFORCE(ctx->HasInput("Grad"),