提交 311ebf1d authored 作者: Frederic's avatar Frederic

If downsample_factor_max_grad_grad use strides, don't use the GPU version

上级 bfa83130
......@@ -1770,7 +1770,9 @@ def local_gpu_downsample_factor_max_grad(node):
node.op.ds == node.op.st):
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 != node.op.ds):
return
x, z, gz = node.inputs
if (x.owner and isinstance(x.owner.op, HostFromGpu)):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论