提交 c1c7efe7 authored 作者: f0k's avatar f0k

GpuCorrMM: Removed pad='auto', added pad='half' and pad='full', added shape…

GpuCorrMM: Removed pad='auto', added pad='half' and pad='full', added shape management for gradients
上级 e5b3c89d
......@@ -1357,7 +1357,7 @@ def local_conv_gemm(node):
pad = (0,0)
if (border_mode == 'full') and (subsample != (1,1)):
# need to simulate this via a padded valid convolution
pad = 'auto'
pad = 'full'
border_mode = 'valid'
if (border_mode == 'valid'):
# need to flip the kernel for valid convolution
......
......@@ -914,7 +914,7 @@ def test_gemm_grads():
i = cuda_tensor4()
k = cuda_tensor4()
pad = 'auto' if mode == 'full' else (0, 0)
pad = 'full' if mode == 'full' else (0, 0)
# TODO: also test custom pad values
corr_op = theano.sandbox.cuda.blas.GpuCorrMM(
'valid', subsample, pad)(i, k)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论