Skip to content

Commit 429d67d

Browse files
authored
Merge pull request #6336 from qingqing01/cudnn_conv_op
Change the type of conv2d in Python API.
2 parents c7e739f + 45aca4e commit 429d67d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/paddle/v2/fluid/layers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -764,7 +764,7 @@ def _get_default_param_initializer():
764764
pre_bias = helper.create_tmp_variable(dtype)
765765

766766
helper.append_op(
767-
type='conv2d',
767+
type='conv2d_cudnn',
768768
inputs={
769769
'Input': input,
770770
'Filter': filter,

0 commit comments

Comments
 (0)