提交 091334f1 authored 作者: Kelvin Xu's avatar Kelvin Xu

change also in GpuDnnConv and GpuDnnConvGradI

上级 904a523d
......@@ -395,7 +395,7 @@ class GpuDnnConv(DnnBase, COp):
img, kerns, desc = inp
top, = grads
top = gpu_contiguous(top)
top = cp_on_negative_strides(top)
d_img = GpuDnnConvGradI()(kerns, top, desc,
img.shape[2], img.shape[3])
......@@ -521,7 +521,7 @@ class GpuDnnConvGradI(DnnBase, COp):
kerns, top, desc, h, w = inp
img, = grads
img = gpu_contiguous(img)
img = cp_on_negative_strides(img)
d_kerns = GpuDnnConvGradW()(img, top, desc,
kerns.shape[2], kerns.shape[3])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论