add paddle_enforce

c8adfb34 · nhzlx · 55334007 · c8adfb34
显示空白变更内容
内联并排

Showing with 2 addition and 0 deletion

paddle/fluid/inference/tensorrt/convert/pool2d_op.cc paddle/fluid/inference/tensorrt/convert/pool2d_op.cc +2 -0

未找到文件。
--- 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 =