提交 581da357 authored 作者: erakra's avatar erakra

avoiding the wrong conv2d op

上级 71000876
......@@ -251,6 +251,8 @@ def local_conv2d_cpu(node):
return None
if node.op.num_groups > 1 or node.op.unshared:
return None
if node.op.filter_dilation != (1, 1):
return None
rval = conv2d(img, kern,
node.op.imshp, node.op.kshp,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论