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

fix optimize_tool bug (#2779)

上级 11716a8b
...@@ -26,9 +26,7 @@ namespace mir { ...@@ -26,9 +26,7 @@ namespace mir {
void ConvBNFusePass::Apply(const std::unique_ptr<SSAGraph>& graph) { void ConvBNFusePass::Apply(const std::unique_ptr<SSAGraph>& graph) {
// initialze fuser params // initialze fuser params
std::vector<bool> conv_has_bias_cases{true, false}; std::vector<bool> conv_has_bias_cases{true, false};
std::vector<std::string> conv_type_cases{ std::vector<std::string> conv_type_cases{"conv2d", "depthwise_conv2d"};
"conv2d", "depthwise_conv2d", "conv2d_transpose"};
// start fuse using params // start fuse using params
for (auto conv_has_bias : conv_has_bias_cases) { for (auto conv_has_bias : conv_has_bias_cases) {
for (auto conv_type : conv_type_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.
先完成此消息的编辑!
想要评论请 注册