提交 ce2d8f86 authored 作者: affanv14's avatar affanv14

change if to elif

上级 f69d0d38
...@@ -2932,7 +2932,7 @@ def local_abstractconv_cudnn_alt(node): ...@@ -2932,7 +2932,7 @@ def local_abstractconv_cudnn_alt(node):
conv_mode=conv_mode, conv_mode=conv_mode,
num_groups=num_groups) num_groups=num_groups)
if isinstance(op, AbstractConv2d_gradWeights): elif isinstance(op, AbstractConv2d_gradWeights):
if(border_mode == 'valid' and subsample == (1, 1) and if(border_mode == 'valid' and subsample == (1, 1) and
filter_dilation == (1, 1) and num_groups == 1): filter_dilation == (1, 1) and num_groups == 1):
img = gpu_contiguous(inp1) img = gpu_contiguous(inp1)
...@@ -2964,7 +2964,7 @@ def local_abstractconv_cudnn_alt(node): ...@@ -2964,7 +2964,7 @@ def local_abstractconv_cudnn_alt(node):
else: else:
return None return None
if isinstance(op, AbstractConv2d_gradInputs): elif isinstance(op, AbstractConv2d_gradInputs):
if border_mode == 'valid' and subsample == (1, 1) and num_groups == 1: if border_mode == 'valid' and subsample == (1, 1) and num_groups == 1:
kerns = gpu_contiguous(inp1.dimshuffle(1, 0, 2, 3)) kerns = gpu_contiguous(inp1.dimshuffle(1, 0, 2, 3))
topgrad = gpu_contiguous(inp2) topgrad = gpu_contiguous(inp2)
...@@ -3036,7 +3036,7 @@ def local_abstractconv3d_cudnn_alt(node): ...@@ -3036,7 +3036,7 @@ def local_abstractconv3d_cudnn_alt(node):
direction_hint=direction_hint, direction_hint=direction_hint,
conv_mode=conv_mode) conv_mode=conv_mode)
if isinstance(op, AbstractConv3d_gradWeights): elif isinstance(op, AbstractConv3d_gradWeights):
if(border_mode == 'valid' and subsample == (1, 1, 1) and if(border_mode == 'valid' and subsample == (1, 1, 1) and
filter_dilation == (1, 1, 1)): filter_dilation == (1, 1, 1)):
img = gpu_contiguous(inp1) img = gpu_contiguous(inp1)
...@@ -3068,7 +3068,7 @@ def local_abstractconv3d_cudnn_alt(node): ...@@ -3068,7 +3068,7 @@ def local_abstractconv3d_cudnn_alt(node):
else: else:
return None return None
if isinstance(op, AbstractConv3d_gradInputs): elif isinstance(op, AbstractConv3d_gradInputs):
if border_mode == 'valid' and subsample == (1, 1, 1): if border_mode == 'valid' and subsample == (1, 1, 1):
kerns = gpu_contiguous(inp1.dimshuffle(1, 0, 2, 3, 4)) kerns = gpu_contiguous(inp1.dimshuffle(1, 0, 2, 3, 4))
topgrad = gpu_contiguous(inp2) topgrad = gpu_contiguous(inp2)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论