提交 7d23f113 authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix local_gpu_downsample_factor_max_grad_grad opt. It should only work for st=(1,1).

上级 1dfceb5d
......@@ -1942,7 +1942,9 @@ def local_gpu_downsample_factor_max_grad_grad(node):
if isinstance(node.op, pool.DownsampleFactorMaxGradGrad):
assert node.op.__props__ == ('ds', 'ignore_border', 'st',
'padding', 'mode')
if node.op.padding != (0, 0) or node.op.mode != 'max':
if (node.op.padding != (0, 0) or
node.op.mode != 'max' or
node.op.st != (1, 1)):
return
x, z, gx = node.inputs
if (x.owner and isinstance(x.owner.op, HostFromGpu)):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论