diff --git a/paddle/fluid/inference/tensorrt/convert/pool2d_op.cc b/paddle/fluid/inference/tensorrt/convert/pool2d_op.cc
index db8e7f8438..2cfd0f6905 100644
--- a/paddle/fluid/inference/tensorrt/convert/pool2d_op.cc
+++ b/paddle/fluid/inference/tensorrt/convert/pool2d_op.cc
@@ -102,8 +102,7 @@ class Pool2dOpConverter : public OpConverter {
       layer->setName(("pool2d (Output: " + output_name + ")").c_str());
       layer->getOutput(0)->setName(output_name.c_str());
       engine_->SetITensor(output_name, layer->getOutput(0));
-      if (test_mode ||
-          output_name == "patch_6_pool1.avg_pool.output.1.tmp_0702") {
+      if (test_mode) {
         engine_->DeclareOutput(output_name);
       }
       return;
@@ -148,8 +147,7 @@ class Pool2dOpConverter : public OpConverter {
     layer->setName(("pool2d (Output: " + output_name + ")").c_str());
     layer->getOutput(0)->setName(output_name.c_str());
     engine_->SetITensor(output_name, layer->getOutput(0));
-    if (test_mode ||
-        output_name == "patch_6_pool1.avg_pool.output.1.tmp_0702") {
+    if (test_mode) {
       engine_->DeclareOutput(output_name);
     }
   }