提交 399911f0 authored 作者: Dustin Webb's avatar Dustin Webb

Fixed bug highlighted by tests.

上级 809a1ab5
...@@ -388,7 +388,7 @@ class GpuDnnConv(DnnBase, COp): ...@@ -388,7 +388,7 @@ class GpuDnnConv(DnnBase, COp):
padh = kh - 1 padh = kh - 1
padw = kw - 1 padw = kw - 1
elif isinstance(desc.border_mode, tuple): elif isinstance(desc.border_mode, tuple):
padh, padw = self.border_mode padh, padw = desc.border_mode
else: else:
assert desc.border_mode == 'valid' assert desc.border_mode == 'valid'
...@@ -559,7 +559,7 @@ class GpuDnnConvGradI(DnnBase, COp): ...@@ -559,7 +559,7 @@ class GpuDnnConvGradI(DnnBase, COp):
padh = shape[0][2] - 1 padh = shape[0][2] - 1
padw = shape[0][3] - 1 padw = shape[0][3] - 1
elif isinstance(desc.border_mode, tuple): elif isinstance(desc.border_mode, tuple):
padh, padw = self.border_mode padh, padw = desc.border_mode
else: else:
assert desc.border_mode == 'valid' assert desc.border_mode == 'valid'
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论