提交 0092ad32 编写于 作者: N nhzlx

delete unused log

上级 329a8c52
......@@ -30,9 +30,7 @@ class ConcatOpConverter : public OpConverter {
framework::OpDesc op_desc(op, nullptr);
// Declare inputs
std::vector<nvinfer1::ITensor*> itensors;
std::cout << "Concat op: " << std::endl;
for (auto& input_name : op_desc.Input("X")) {
std::cout << input_name << std::endl;
itensors.push_back(engine_->GetITensor(input_name));
}
int axis = boost::get<int>(op_desc.GetAttr("axis"));
......
......@@ -26,9 +26,6 @@ class Conv2dOpConverter : public OpConverter {
<< "convert a fluid conv2d op to tensorrt conv layer without bias";
framework::OpDesc op_desc(op, nullptr);
std::cout << "Conv op: " << std::endl;
std::cout << op_desc.Input("Input").front() << std::endl;
std::cout << op_desc.Output("Output").front() << std::endl;
PADDLE_ENFORCE_EQ(op_desc.Input("Input").size(), 1);
PADDLE_ENFORCE_EQ(op_desc.Input("Filter").size(), 1); // Y is a weight
PADDLE_ENFORCE_EQ(op_desc.Output("Output").size(), 1);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册