Skip to content

Commit 8f9a8c4

Browse files
committed
delete unused test code.
test=develop
1 parent 83f8c40 commit 8f9a8c4

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

paddle/fluid/inference/tensorrt/convert/pool2d_op.cc

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -102,8 +102,7 @@ class Pool2dOpConverter : public OpConverter {
102102
layer->setName(("pool2d (Output: " + output_name + ")").c_str());
103103
layer->getOutput(0)->setName(output_name.c_str());
104104
engine_->SetITensor(output_name, layer->getOutput(0));
105-
if (test_mode ||
106-
output_name == "patch_6_pool1.avg_pool.output.1.tmp_0702") {
105+
if (test_mode) {
107106
engine_->DeclareOutput(output_name);
108107
}
109108
return;
@@ -148,8 +147,7 @@ class Pool2dOpConverter : public OpConverter {
148147
layer->setName(("pool2d (Output: " + output_name + ")").c_str());
149148
layer->getOutput(0)->setName(output_name.c_str());
150149
engine_->SetITensor(output_name, layer->getOutput(0));
151-
if (test_mode ||
152-
output_name == "patch_6_pool1.avg_pool.output.1.tmp_0702") {
150+
if (test_mode) {
153151
engine_->DeclareOutput(output_name);
154152
}
155153
}

0 commit comments

Comments
 (0)