未验证 提交 d021095e 编写于 作者: L Leo Chen 提交者: GitHub

add conv2d and conv2d_grad as default deny cinn ops (#51645)

上级 543da561
......@@ -79,7 +79,8 @@ class OpTransInfo {
DeParamCondT deny_param_cond_{{"batch_norm", {"ReserveSpace"}},
{"batch_norm_grad", {"ReserveSpace"}}};
std::unordered_set<std::string> default_deny_ops_{"feed", "fetch"};
std::unordered_set<std::string> default_deny_ops_{
"feed", "fetch", "conv2d", "conv2d_grad"};
};
// A pass named BuildCinnPass, the function of this pass is:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册