提交 407c7766 authored 作者: Gijs van Tulder's avatar Gijs van Tulder

Rename opt local_conv2d_gpu_conv.

上级 256a3637
......@@ -2733,7 +2733,7 @@ optdb.register('local_inplace_gpu_sparse_block_outer',
AbstractConv3d,
AbstractConv3d_gradWeights,
AbstractConv3d_gradInputs])
def local_conv2d_gpu_conv(node):
def local_conv_gpu_conv(node):
"""
gpu_from_host(AbstractConv) -> AbstractConv(gpu_from_host)
......@@ -2785,7 +2785,7 @@ def local_conv2d_gpu_conv(node):
return [tensor.as_tensor_variable(out)]
else:
return [out]
register_opt()(local_conv2d_gpu_conv)
register_opt()(local_conv_gpu_conv)
# Corrmm opt
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论