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