提交 332601b4 authored 作者: Frederic Bastien's avatar Frederic Bastien

fix gh-4036. We need the shape and we don't compute it. It wasn't timmed, so…

fix gh-4036. We need the shape and we don't compute it. It wasn't timmed, so don't try this optimization.
上级 dbb49e4b
...@@ -2770,7 +2770,8 @@ def local_abstractconv_gemm(node): ...@@ -2770,7 +2770,8 @@ def local_abstractconv_gemm(node):
(node.op.imshp is not None) and (node.op.imshp is not None) and
(None not in node.op.imshp[-2:]) and (None not in node.op.imshp[-2:]) and
(node.op.kshp is not None) and (node.op.kshp is not None) and
(None not in node.op.kshp)): (None not in node.op.kshp) and
border_mode != "half"):
# we know the kernel and output size # we know the kernel and output size
prod1 = node.op.kshp[0] * node.op.kshp[1] prod1 = node.op.kshp[0] * node.op.kshp[1]
prod2 = ((node.op.imshp[-2] - node.op.kshp[0] + 1) * prod2 = ((node.op.imshp[-2] - node.op.kshp[0] + 1) *
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论