提交 cc3d9492 编写于 作者: Z zlsh80826

remove slice, scale op, will add later

上级 6e3255a1
......@@ -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,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册