Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 cc3d9492 编写于 作者: Z zlsh80826
浏览文件
操作

remove slice, scale op, will add later

上级 6e3255a1
隐藏空白更改
内联 并排
Showing with 32 addition and 29 deletion
paddle/fluid/inference/tensorrt/op_teller.cc
浏览文件 @ cc3d9492
...@@ -54,35 +54,38 @@ struct SimpleOpTypeSetTeller : public Teller { ...@@ -54,35 +54,38 @@ struct SimpleOpTypeSetTeller : public Teller {
"leaky_relu", "leaky_relu",
"fc", "fc",
"relu6", "relu6",
"concat"}; "concat",
std::unordered_set<std::string> teller_set{"mul", "scale",
"conv2d", "elementwise_mul",
"pool2d", "conv2d_transpose"};
"relu", std::unordered_set<std::string> teller_set{
"softmax", "mul",
"sigmoid", "conv2d",
"hard_swish", "pool2d",
"depthwise_conv2d", "relu",
"batch_norm", "softmax",
"concat", "sigmoid",
"tanh", "hard_swish",
"pad", "depthwise_conv2d",
"elementwise_add", "batch_norm",
"elementwise_mul", "concat",
"dropout", "tanh",
"prelu", "pad",
"conv2d_transpose", "elementwise_add",
"leaky_relu", "elementwise_mul",
"fc", "dropout",
"shuffle_channel", "prelu",
"swish", "conv2d_transpose",
"split", "leaky_relu",
"instance_norm", "fc",
"gelu", "shuffle_channel",
"layer_norm", "swish",
"scale", "split",
"slice", "instance_norm",
"stack"}; "gelu",
"layer_norm",
"scale",
};
}; };
bool OpTeller::Tell(const std::string& op_type, const framework::OpDesc& desc, bool OpTeller::Tell(const std::string& op_type, const framework::OpDesc& desc,
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或