From fbe15e642d1209d00aef686f0e364b77067f7b1b Mon Sep 17 00:00:00 2001
From: jiangjiajun <jiangjiajun@baidu.com>
Date: Wed, 14 Aug 2019 18:00:51 +0800
Subject: [PATCH] optimize for depthwise conv2d

---
 x2paddle/op_mapper/tf_op_mapper.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/x2paddle/op_mapper/tf_op_mapper.py b/x2paddle/op_mapper/tf_op_mapper.py
index 2136604..ba33d6d 100644
--- a/x2paddle/op_mapper/tf_op_mapper.py
+++ b/x2paddle/op_mapper/tf_op_mapper.py
@@ -503,6 +503,7 @@ class TFOpMapper(OpMapper):
             "stride": strides[2:4],
             "dilation": dilations[2:4],
             "groups": k_size[3] * in_shape[1],
+            "use_cudnn": False,
             "padding": padding
         }
         node.fluid_code.add_layer("conv2d",
-- 
GitLab