File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed
paddle/fluid/inference/tensorrt/convert Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff line change @@ -102,8 +102,7 @@ class Pool2dOpConverter : public OpConverter {
102
102
layer->setName ((" pool2d (Output: " + output_name + " )" ).c_str ());
103
103
layer->getOutput (0 )->setName (output_name.c_str ());
104
104
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) {
107
106
engine_->DeclareOutput (output_name);
108
107
}
109
108
return ;
@@ -148,8 +147,7 @@ class Pool2dOpConverter : public OpConverter {
148
147
layer->setName ((" pool2d (Output: " + output_name + " )" ).c_str ());
149
148
layer->getOutput (0 )->setName (output_name.c_str ());
150
149
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) {
153
151
engine_->DeclareOutput (output_name);
154
152
}
155
153
}
You can’t perform that action at this time.
0 commit comments