Skip to content

P
PaddleDetection

s920243400 / PaddleDetection
与 Fork 源项目一致

Fork自 PaddlePaddle / PaddleDetection

通知 2
Star 0
Fork 0
P
PaddleDetection
提交 c8adfb34 编写于 作者: N nhzlx
浏览文件
操作

add paddle_enforce

上级 55334007
显示空白变更内容
内联 并排
Showing with 2 addition and 0 deletion
paddle/fluid/inference/tensorrt/convert/pool2d_op.cc
浏览文件 @ c8adfb34
...@@ -29,6 +29,8 @@ class Pool2dOpConverter : public OpConverter { ...@@ -29,6 +29,8 @@ class Pool2dOpConverter : public OpConverter {
<< "convert a fluid pool2d op to tensorrt pool2d layer without bias"; << "convert a fluid pool2d op to tensorrt pool2d layer without bias";
framework::OpDesc op_desc(op, nullptr); framework::OpDesc op_desc(op, nullptr);
// Declare inputs // 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]); auto* input1 = engine_->GetITensor(op_desc.Input("X")[0]);
std::string pool_type = std::string pool_type =
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或