Merge pull request #6336 from qingqing01/cudnn_conv_op

Change the type of conv2d in Python API.
8 years ago · 429d67deee
parent c7e739f542 45aca4e9bb
commit 429d67deee
1 changed files with 1 additions and 1 deletions
--- 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,