diff --git a/paddle/fluid/inference/tensorrt/op_teller.cc b/paddle/fluid/inference/tensorrt/op_teller.cc index 5f4398895b387444c5e355db0cf611589d0fc176..70ead9720d2ebcb15ae0173dc0ba7c2095a4f4d4 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 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 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,