提交 816cdaf6 authored 作者: affanv14's avatar affanv14 提交者: Mohammed Affan

skip old cpu convolution for grouped convolution

上级 ffa7a7ba
......@@ -242,6 +242,8 @@ def local_conv2d_cpu(node):
if not node.op.filter_flip:
# Not tested yet
return None
if node.op.num_groups > 1:
return None
rval = conv2d(img, kern,
node.op.imshp, node.op.kshp,
......@@ -299,6 +301,8 @@ def local_conv2d_gradweight_cpu(node):
if not node.op.filter_flip:
# Not tested yet
return
if node.op.num_groups > 1:
return None
if node.op.border_mode == 'valid' and \
(node.op.subsample != (1, 1)):
......@@ -451,6 +455,8 @@ def local_conv2d_gradinputs_cpu(node):
if not node.op.filter_flip:
# Not tested yet
return None
if node.op.num_groups > 1:
return None
# Conv 3d implementation, needed when subsample > 2
if node.op.border_mode == 'valid' and node.op.subsample != (1, 1):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论