remove slice, scale op, will add later

cc3d9492 · zlsh80826 · 6e3255a1 · cc3d9492
显示空白变更内容
内联并排

Showing with 32 addition and 29 deletion

paddle/fluid/inference/tensorrt/op_teller.cc paddle/fluid/inference/tensorrt/op_teller.cc +32 -29

未找到文件。
--- a/paddle/fluid/inference/tensorrt/op_teller.cc
+++ b/paddle/fluid/inference/tensorrt/op_teller.cc
@@ -54,8 +54,12 @@ struct SimpleOpTypeSetTeller : public Teller {
                                                  "leaky_relu",
                                                  "fc",
                                                  "relu6",
-                                                  "concat"};
-  std::unordered_set<std::string> teller_set{"mul",
+                                                  "concat",
+                                                  "scale",
+                                                  "elementwise_mul",
+                                                  "conv2d_transpose"};
+  std::unordered_set<std::string> teller_set{
+      "mul",
      "conv2d",
      "pool2d",
      "relu",
@@ -81,8 +85,7 @@ struct SimpleOpTypeSetTeller : public Teller {
      "gelu",
      "layer_norm",
      "scale",
-                                             "slice",
-                                             "stack"};
+  };
 };

 bool OpTeller::Tell(const std::string& op_type, const framework::OpDesc& desc,