Skip to content

Commit c8adfb3

Browse files
committed
add paddle_enforce
1 parent 5533400 commit c8adfb3

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

paddle/fluid/inference/tensorrt/convert/pool2d_op.cc

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,8 @@ class Pool2dOpConverter : public OpConverter {
2929
<< "convert a fluid pool2d op to tensorrt pool2d layer without bias";
3030
framework::OpDesc op_desc(op, nullptr);
3131
// Declare inputs
32+
PADDLE_ENFORCE_EQ(op_desc.Input("X").size(), 1);
33+
PADDLE_ENFORCE_EQ(op_desc.Output("Out").size(), 1);
3234
auto* input1 = engine_->GetITensor(op_desc.Input("X")[0]);
3335

3436
std::string pool_type =

0 commit comments

Comments
 (0)