We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5533400 commit c8adfb3Copy full SHA for c8adfb3
paddle/fluid/inference/tensorrt/convert/pool2d_op.cc
@@ -29,6 +29,8 @@ class Pool2dOpConverter : public OpConverter {
29
<< "convert a fluid pool2d op to tensorrt pool2d layer without bias";
30
framework::OpDesc op_desc(op, nullptr);
31
// Declare inputs
32
+ PADDLE_ENFORCE_EQ(op_desc.Input("X").size(), 1);
33
+ PADDLE_ENFORCE_EQ(op_desc.Output("Out").size(), 1);
34
auto* input1 = engine_->GetITensor(op_desc.Input("X")[0]);
35
36
std::string pool_type =
0 commit comments