提交 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
......@@ -2767,10 +2767,11 @@ def local_abstractconv_gemm(node):
# GpuConv does not always store information on the batchsize and
# channels, though, so we only use what information we have.)
if ((subsample == (1, 1)) and
(node.op.imshp is not None) and
(None not in node.op.imshp[-2:]) and
(node.op.kshp is not None) and
(None not in node.op.kshp)):
(node.op.imshp is not None) and
(None not in node.op.imshp[-2:]) and
(node.op.kshp is not None) and
(None not in node.op.kshp) and
border_mode != "half"):
# we know the kernel and output size
prod1 = node.op.kshp[0] * node.op.kshp[1]
prod2 = ((node.op.imshp[-2] - node.op.kshp[0] + 1) *
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论