Merge pull request #6336 from qingqing01/cudnn_conv_op

Change the type of conv2d in Python API.

Merge pull request #6336 from qingqing01/cudnn_conv_op
Change the type of conv2d in Python API.
429d67de · qingqing01 · GitHub · c7e739f5 · 45aca4e9 · 429d67de
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

python/paddle/v2/fluid/layers.py python/paddle/v2/fluid/layers.py +1 -1

未找到文件。
--- a/python/paddle/v2/fluid/layers.py
+++ b/python/paddle/v2/fluid/layers.py
@@ -764,7 +764,7 @@ def conv2d(input,
    pre_bias = helper.create_tmp_variable(dtype)

    helper.append_op(
-        type='conv2d',
+        type='conv2d_cudnn',
        inputs={
            'Input': input,
            'Filter': filter,