diff --git a/paddle/operators/linear_chain_crf_op.cc b/paddle/operators/linear_chain_crf_op.cc
index 434382a72fe580ca2793b618931f60206aa131b8..fd473980654ecb58a39c9bc0675a7af378658d01 100644
--- a/paddle/operators/linear_chain_crf_op.cc
+++ b/paddle/operators/linear_chain_crf_op.cc
@@ -119,7 +119,7 @@ class LinearChainCrfOp : public framework::OperatorWithKernel {
   using framework::OperatorWithKernel::OperatorWithKernel;
 
  protected:
-  void InferShape(framework::InferShapeContextBase* ctx) const override {}
+  void InferShape(framework::InferShapeContext* ctx) const override {}
 };
 
 class LinearChainCrfGradOp : public framework::OperatorWithKernel {
@@ -127,7 +127,7 @@ class LinearChainCrfGradOp : public framework::OperatorWithKernel {
   using framework::OperatorWithKernel::OperatorWithKernel;
 
  protected:
-  void InferShape(framework::InferShapeContextBase* ctx) const override {}
+  void InferShape(framework::InferShapeContext* ctx) const override {}
 };
 
 }  // namespace operators