未验证 提交 6b3ffe99 编写于 作者: J JingZhuangzhuang 提交者: GitHub

cherry_pick conv2d omission (#37419)

上级 9ffb43be
...@@ -184,13 +184,6 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8, ...@@ -184,13 +184,6 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8,
return false; return false;
for (auto& teller : tellers_) { for (auto& teller : tellers_) {
if (op_type == "depthwise_conv2d") {
std::vector<int> paddings =
BOOST_GET_CONST(std::vector<int>, desc.GetAttr("paddings"));
if (paddings.size() > 2) return false;
}
if (op_type == "relu" || op_type == "relu6" || op_type == "tanh" || if (op_type == "relu" || op_type == "relu6" || op_type == "tanh" ||
op_type == "sigmoid") { op_type == "sigmoid") {
auto* block = desc.Block(); auto* block = desc.Block();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册