File tree Expand file tree Collapse file tree 2 files changed +0
-5
lines changed
paddle/fluid/inference/tensorrt/convert Expand file tree Collapse file tree 2 files changed +0
-5
lines changed Original file line number Diff line number Diff line change @@ -30,9 +30,7 @@ class ConcatOpConverter : public OpConverter {
30
30
framework::OpDesc op_desc (op, nullptr );
31
31
// Declare inputs
32
32
std::vector<nvinfer1::ITensor*> itensors;
33
- std::cout << " Concat op: " << std::endl;
34
33
for (auto & input_name : op_desc.Input (" X" )) {
35
- std::cout << input_name << std::endl;
36
34
itensors.push_back (engine_->GetITensor (input_name));
37
35
}
38
36
int axis = boost::get<int >(op_desc.GetAttr (" axis" ));
Original file line number Diff line number Diff line change @@ -26,9 +26,6 @@ class Conv2dOpConverter : public OpConverter {
26
26
<< " convert a fluid conv2d op to tensorrt conv layer without bias" ;
27
27
28
28
framework::OpDesc op_desc (op, nullptr );
29
- std::cout << " Conv op: " << std::endl;
30
- std::cout << op_desc.Input (" Input" ).front () << std::endl;
31
- std::cout << op_desc.Output (" Output" ).front () << std::endl;
32
29
PADDLE_ENFORCE_EQ (op_desc.Input (" Input" ).size (), 1 );
33
30
PADDLE_ENFORCE_EQ (op_desc.Input (" Filter" ).size (), 1 ); // Y is a weight
34
31
PADDLE_ENFORCE_EQ (op_desc.Output (" Output" ).size (), 1 );
You can’t perform that action at this time.
0 commit comments