提交 8318ac85 authored 作者: Gijs van Tulder's avatar Gijs van Tulder

Fix padding checks for pooling in cuda/opt.

上级 f9aa6177
......@@ -1922,7 +1922,7 @@ def local_gpu_downsample_factor_max(node):
return
ws, stride, pad = ret
if (nd != 2 or
max(node.op.padding) != 0 or
max(pad) != 0 or
node.op.mode != 'max' or
stride != ws):
return
......@@ -1949,7 +1949,7 @@ def local_gpu_downsample_factor_max_grad(node):
return
ws, stride, pad = ret
if (nd != 2 or
max(node.op.padding) != 0 or
max(pad) != 0 or
node.op.mode != 'max' or
stride != ws):
return
......@@ -1980,7 +1980,7 @@ def local_gpu_downsample_factor_max_grad_grad(node):
return
ws, stride, pad = ret
if (nd != 2 or
max(node.op.padding) != 0 or
max(pad) != 0 or
node.op.mode != 'max' or
stride != ws):
return
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论