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

Fixed bug highlighted by tests.

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