m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

delete unused test code.

Browse Source 
test=develop
panyx0718-patch-1
nhzlx 6 years ago
parent 83f8c403a7
commit 8f9a8c455a
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File

6
paddle/fluid/inference/tensorrt/convert/pool2d_op.cc
Unescape View File

@ -102,8 +102,7 @@ class Pool2dOpConverter : public OpConverter {
layer->setName(("pool2d (Output: " + output_name + ")").c_str()); layer->setName(("pool2d (Output: " + output_name + ")").c_str());
layer->getOutput(0)->setName(output_name.c_str()); layer->getOutput(0)->setName(output_name.c_str());
engine_->SetITensor(output_name, layer->getOutput(0)); engine_->SetITensor(output_name, layer->getOutput(0));
if (test_mode || if (test_mode) {
output_name == "patch_6_pool1.avg_pool.output.1.tmp_0702") {
engine_->DeclareOutput(output_name); engine_->DeclareOutput(output_name);
} }
return; return;
@ -148,8 +147,7 @@ class Pool2dOpConverter : public OpConverter {
layer->setName(("pool2d (Output: " + output_name + ")").c_str()); layer->setName(("pool2d (Output: " + output_name + ")").c_str());
layer->getOutput(0)->setName(output_name.c_str()); layer->getOutput(0)->setName(output_name.c_str());
engine_->SetITensor(output_name, layer->getOutput(0)); engine_->SetITensor(output_name, layer->getOutput(0));
if (test_mode || if (test_mode) {
output_name == "patch_6_pool1.avg_pool.output.1.tmp_0702") {
engine_->DeclareOutput(output_name); engine_->DeclareOutput(output_name);
} }
} }

Write Preview
Loading…
Cancel
Save