提交 0ecc5689 authored 作者: affanv14's avatar affanv14

return None if conditions are not satisfied

上级 3a333931
......@@ -2961,6 +2961,8 @@ def local_abstractconv_cudnn_alternative(node):
conv = conv[:, :, ::-1, ::-1]
rval = as_gpuarray_variable(conv.dimshuffle(1, 0, 2, 3), ctx_name)
else:
return None
if isinstance(op, AbstractConv2d_gradInputs):
if border_mode == 'valid' and subsample == (1, 1) and num_groups == 1:
......@@ -2985,6 +2987,8 @@ def local_abstractconv_cudnn_alternative(node):
shape = assert_conv_shape(shape)
out = GpuAllocEmpty(dtype=topgrad.dtype, context_name=ctx_name)(*shape)
rval = GpuDnnConv(algo=None, num_groups=num_groups)(topgrad, kerns, out, desc)
else:
return None
return [rval]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论