未验证 提交 d42a1dc3 编写于 作者: Z zhoutianzi666 提交者: GitHub

[Paddle-TRT] Fix conv2d (#47034)

* forbid Conv2D into Paddle-TensoRT when weight is not persistable.
上级 281891c5
......@@ -280,6 +280,15 @@ struct SimpleOpTypeSetTeller : public Teller {
}
}
#endif
auto* block = desc.Block();
if (block) {
auto* filter_var_desc = block->FindVar(desc.Input("Filter")[0]);
if (!filter_var_desc->Persistable()) {
VLOG(3) << "Trt not support filter is a intermediate tensor in "
"conv2d op.";
return false;
}
}
}
if (op_type == "deformable_conv") {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册