提交 3a333931 authored 作者: affanv14's avatar affanv14

add few conditions for cudnn fwd alternative

上级 3d8e5a5e
...@@ -2915,9 +2915,11 @@ def local_abstractconv_cudnn_alternative(node): ...@@ -2915,9 +2915,11 @@ def local_abstractconv_cudnn_alternative(node):
conv_mode = 'cross' conv_mode = 'cross'
if isinstance(op, AbstractConv2d): if isinstance(op, AbstractConv2d):
if border_mode == 'half' or subsample != (1, 1) or num_groups != 1:
return None
if border_mode == 'full': if border_mode == 'full':
direction_hint = 'bprop inputs' direction_hint = 'bprop inputs'
elif border_mode == 'valid': elif border_mode == 'valid' and filter_dilation == (1, 1):
direction_hint = 'bprop weights' direction_hint = 'bprop weights'
else: else:
return None return None
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论