add conv2d transpose fuse (#1909)

57ee8714 · tensor-tang · GitHub · de43e479 · 57ee8714
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

lite/core/mir/fusion/conv_elementwise_fuse_pass.cc lite/core/mir/fusion/conv_elementwise_fuse_pass.cc +3 -0

未找到文件。
--- a/lite/core/mir/fusion/conv_elementwise_fuse_pass.cc
+++ b/lite/core/mir/fusion/conv_elementwise_fuse_pass.cc
@@ -28,6 +28,9 @@ void ConvElementwiseFusePass::Apply(const std::unique_ptr<SSAGraph>& graph) {

  fusion::ConvElementwiseFuser depthwise_fuser("depthwise_conv2d");
  depthwise_fuser(graph.get());
+
+  fusion::ConvElementwiseFuser conv2d_transpose_fuser("conv2d_transpose");
+  conv2d_transpose_fuser(graph.get());
 }

 }  // namespace mir