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