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

forbid conv op whose weight is not a persistable weight into Paddle-TRT (#48763)

上级 3cb8db8f
......@@ -284,6 +284,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.
先完成此消息的编辑!
想要评论请 注册