From 6b3ffe997706acdea9a3230cd0a82ba74e13db9d Mon Sep 17 00:00:00 2001 From: JingZhuangzhuang <75348594+JZZ-NOTE@users.noreply.github.com> Date: Tue, 23 Nov 2021 10:01:26 +0800 Subject: [PATCH] cherry_pick conv2d omission (#37419) --- paddle/fluid/inference/tensorrt/op_teller.cc | 7 ------- 1 file changed, 7 deletions(-) diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc index e623167a1f2..9d1831bffb0 100644 --- a/paddle/fluid/inference/tensorrt/op_teller.cc +++ b/paddle/fluid/inference/tensorrt/op_teller.cc @@ -184,13 +184,6 @@ bool OpTeller::Tell(const framework::ir::Node* node, bool use_no_calib_int8, return false; for (auto& teller : tellers_) { - if (op_type == "depthwise_conv2d") { - std::vector paddings = - BOOST_GET_CONST(std::vector, desc.GetAttr("paddings")); - - if (paddings.size() > 2) return false; - } - if (op_type == "relu" || op_type == "relu6" || op_type == "tanh" || op_type == "sigmoid") { auto* block = desc.Block(); -- GitLab