add paddle_enforce

7 years ago · c8adfb3451
parent 5533400720
commit c8adfb3451
1 changed files with 2 additions and 0 deletions
--- a/paddle/fluid/inference/tensorrt/convert/pool2d_op.cc
+++ b/paddle/fluid/inference/tensorrt/convert/pool2d_op.cc
@ -29,6 +29,8 @@ class Pool2dOpConverter : public OpConverter {
        << "convert a fluid pool2d op to tensorrt pool2d layer without bias";
    framework::OpDesc op_desc(op, nullptr);
    // Declare inputs
+    PADDLE_ENFORCE_EQ(op_desc.Input("X").size(), 1);
+    PADDLE_ENFORCE_EQ(op_desc.Output("Out").size(), 1);
    auto* input1 = engine_->GetITensor(op_desc.Input("X")[0]);

    std::string pool_type =