From cc3d9492a16f4acb2d751eaa7186c5f37cb54939 Mon Sep 17 00:00:00 2001
From: zlsh80826 <zlsh80826@gmail.com>
Date: Tue, 4 Aug 2020 18:08:37 +0800
Subject: [PATCH] remove slice, scale op, will add later

---
 paddle/fluid/inference/tensorrt/op_teller.cc | 61 ++++++++++----------
 1 file changed, 32 insertions(+), 29 deletions(-)

diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc
index 5f4398895b3..70ead9720d2 100644
--- a/paddle/fluid/inference/tensorrt/op_teller.cc
+++ b/paddle/fluid/inference/tensorrt/op_teller.cc
@@ -54,35 +54,38 @@ struct SimpleOpTypeSetTeller : public Teller {
                                                   "leaky_relu",
                                                   "fc",
                                                   "relu6",
-                                                  "concat"};
-  std::unordered_set<std::string> teller_set{"mul",
-                                             "conv2d",
-                                             "pool2d",
-                                             "relu",
-                                             "softmax",
-                                             "sigmoid",
-                                             "hard_swish",
-                                             "depthwise_conv2d",
-                                             "batch_norm",
-                                             "concat",
-                                             "tanh",
-                                             "pad",
-                                             "elementwise_add",
-                                             "elementwise_mul",
-                                             "dropout",
-                                             "prelu",
-                                             "conv2d_transpose",
-                                             "leaky_relu",
-                                             "fc",
-                                             "shuffle_channel",
-                                             "swish",
-                                             "split",
-                                             "instance_norm",
-                                             "gelu",
-                                             "layer_norm",
-                                             "scale",
-                                             "slice",
-                                             "stack"};
+                                                  "concat",
+                                                  "scale",
+                                                  "elementwise_mul",
+                                                  "conv2d_transpose"};
+  std::unordered_set<std::string> teller_set{
+      "mul",
+      "conv2d",
+      "pool2d",
+      "relu",
+      "softmax",
+      "sigmoid",
+      "hard_swish",
+      "depthwise_conv2d",
+      "batch_norm",
+      "concat",
+      "tanh",
+      "pad",
+      "elementwise_add",
+      "elementwise_mul",
+      "dropout",
+      "prelu",
+      "conv2d_transpose",
+      "leaky_relu",
+      "fc",
+      "shuffle_channel",
+      "swish",
+      "split",
+      "instance_norm",
+      "gelu",
+      "layer_norm",
+      "scale",
+  };
 };
 
 bool OpTeller::Tell(const std::string& op_type, const framework::OpDesc& desc,
-- 
GitLab