提交 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):
@local_optimizer([GpuConv])
def local_conv_gemm(node):
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)):
img, kern = node.inputs
img = gpu_contiguous(img)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论