diff --git a/paddle/fluid/operators/gather_tree_op.cc b/paddle/fluid/operators/gather_tree_op.cc
index 26f9989121e9eb32d27817d23fa972e5e8fcf1b7..f743727a2814401356cf74e5125fe950e0973971 100644
--- a/paddle/fluid/operators/gather_tree_op.cc
+++ b/paddle/fluid/operators/gather_tree_op.cc
@@ -22,18 +22,19 @@ class GatherTreeOp : public framework::OperatorWithKernel {
   using framework::OperatorWithKernel::OperatorWithKernel;
 
   void InferShape(framework::InferShapeContext* ctx) const override {
-    PADDLE_ENFORCE(ctx->HasInput("Ids"),
-                   "Input(Ids) of GatherTreeOp should not be null.");
-    PADDLE_ENFORCE(ctx->HasInput("Parents"),
-                   "Input(Parents) of GatherTreeOp should not be null.");
-    PADDLE_ENFORCE(ctx->HasOutput("Out"),
-                   "Output(Out) of GatherTreeOp should not be null.");
+    PADDLE_ENFORCE_EQ(ctx->HasInput("Ids"), true,
+                      "Input(Ids) of GatherTreeOp should not be null.");
+    PADDLE_ENFORCE_EQ(ctx->HasInput("Parents"), true,
+                      "Input(Parents) of GatherTreeOp should not be null.");
+    PADDLE_ENFORCE_EQ(ctx->HasOutput("Out"), true,
+                      "Output(Out) of GatherTreeOp should not be null.");
 
     auto ids_dims = ctx->GetInputDim("Ids");
     auto parents_dims = ctx->GetInputDim("Parents");
-    PADDLE_ENFORCE(ids_dims == parents_dims,
-                   "The shape of Input(Parents) must be same with the shape of "
-                   "Input(Ids).");
+    PADDLE_ENFORCE_EQ(
+        ids_dims == parents_dims, true,
+        "The shape of Input(Parents) must be same with the shape of "
+        "Input(Ids).");
     ctx->SetOutputDim("Out", ids_dims);
   }