提交 45aca4e9 编写于 作者: D dangqingqing

Change the type of conv2d in Python API.

上级 229c2e78
...@@ -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.
先完成此消息的编辑!
想要评论请 注册