提交 124c43a0 编写于 作者: H HappyAngel 提交者: yiicy

fix optimize_tool bug (#2779)

上级 11716a8b
......@@ -26,9 +26,7 @@ namespace mir {
void ConvBNFusePass::Apply(const std::unique_ptr<SSAGraph>& graph) {
// initialze fuser params
std::vector<bool> conv_has_bias_cases{true, false};
std::vector<std::string> conv_type_cases{
"conv2d", "depthwise_conv2d", "conv2d_transpose"};
std::vector<std::string> conv_type_cases{"conv2d", "depthwise_conv2d"};
// start fuse using params
for (auto conv_has_bias : conv_has_bias_cases) {
for (auto conv_type : conv_type_cases) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册