提交 a1296c66 authored 作者: Kashif Rasul's avatar Kashif Rasul

fix shape mismatch in GpuDnnPoolGrad

fixes for issue #3347
上级 450dfcb2
......@@ -2321,10 +2321,10 @@ if True:
(inp_grad.owner and isinstance(inp_grad.owner.op,
HostFromGpu))):
desc = GpuDnnPoolDesc(ws=ds, stride=st, mode=mode, pad=pad)()
contiguous_inp_grad = gpu_contiguous(inp_grad)
ret = GpuDnnPoolGrad()(gpu_contiguous(inp),
gpu_contiguous(numpy.empty((1,1,1,1),
dtype=numpy.float32)),
gpu_contiguous(inp_grad),
contiguous_inp_grad,
contiguous_inp_grad,
desc)
return [host_from_gpu(ret)]
......
......@@ -1763,9 +1763,10 @@ def local_avg_pool_dnn_grad_stride(node):
mode = node.op.mode
desc = GpuDnnPoolDesc(ws=ds, stride=st, mode=mode, pad=pad)()
contiguous_inp_grad = gpu_contiguous(inp_grad)
return GpuDnnPoolGrad()(gpu_contiguous(inp),
gpu_contiguous(numpy.empty((1, 1, 1, 1), dtype=numpy.float32)),
gpu_contiguous(inp_grad),
contiguous_inp_grad,
contiguous_inp_grad,
desc)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论