From 8f9a8c455a2ec22f5f67cc464d5b6a82cafbfb57 Mon Sep 17 00:00:00 2001 From: nhzlx Date: Fri, 16 Nov 2018 08:14:04 +0000 Subject: [PATCH] delete unused test code. test=develop --- paddle/fluid/inference/tensorrt/convert/pool2d_op.cc | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/paddle/fluid/inference/tensorrt/convert/pool2d_op.cc b/paddle/fluid/inference/tensorrt/convert/pool2d_op.cc index db8e7f843..2cfd0f690 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); } } -- GitLab