未验证 提交 429d67de 编写于 作者: Q qingqing01 提交者: GitHub

Merge pull request #6336 from qingqing01/cudnn_conv_op

Change the type of conv2d in Python API.
...@@ -764,7 +764,7 @@ def conv2d(input, ...@@ -764,7 +764,7 @@ def conv2d(input,
pre_bias = helper.create_tmp_variable(dtype) pre_bias = helper.create_tmp_variable(dtype)
helper.append_op( helper.append_op(
type='conv2d', type='conv2d_cudnn',
inputs={ inputs={
'Input': input, 'Input': input,
'Filter': filter, 'Filter': filter,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册