提交 74ea01ac authored 作者: Arjun Jain's avatar Arjun Jain

fixed the bug after which test_full also passes all tests.

上级 497a2d9f
...@@ -1286,7 +1286,7 @@ def local_gpu_downsample_factor_max_grad(node): ...@@ -1286,7 +1286,7 @@ def local_gpu_downsample_factor_max_grad(node):
@local_optimizer([GpuConv]) @local_optimizer([GpuConv])
def local_conv_gemm(node): def local_conv_gemm(node):
if (isinstance(node.op, GpuConv) and if (isinstance(node.op, GpuConv) and
node.op.border_mode == 'valid' and node.op.border_mode in ['full', 'valid'] and
node.op.subsample == (1, 1)): node.op.subsample == (1, 1)):
img, kern = node.inputs img, kern = node.inputs
img = gpu_contiguous(img) img = gpu_contiguous(img)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论