optimize for depthwise conv2d

fbe15e64 · jiangjiajun · e1fa0be6 · fbe15e64
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

x2paddle/op_mapper/tf_op_mapper.py x2paddle/op_mapper/tf_op_mapper.py +1 -0

未找到文件。
--- 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",