提交 6babe389 authored 作者: --global's avatar --global

Fix typos in grad methods of CuDNN 3d convolution

上级 906f4eac
......@@ -588,8 +588,8 @@ class GpuDnnConv3d(GpuDnnConv):
top = gpu_contiguous(top)
d_img = GpuDnnConvGrad3dI()(kerns, top, gpu_alloc_empty(*img.shape), desc)
d_kerns = GpuDnnConvGrad3dW()(img, top, gpu_alloc_empty(*kerns.shape), desc)
d_img = GpuDnnConv3dGradI()(kerns, top, gpu_alloc_empty(*img.shape), desc)
d_kerns = GpuDnnConv3dGradW()(img, top, gpu_alloc_empty(*kerns.shape), desc)
d_alpha = grad_not_implemented(self, 4, alpha)
d_beta = grad_not_implemented(self, 5, beta)
......@@ -754,7 +754,7 @@ class GpuDnnConv3dGradW(GpuDnnConvGradW):
kerns = gpu_contiguous(kerns)
d_img = GpuDnnConvGrad3dI()(kerns, top, gpu_alloc_empty(*img.shape), desc)
d_img = GpuDnnConv3dGradI()(kerns, top, gpu_alloc_empty(*img.shape), desc)
d_top = GpuDnnConv3d()(img, kerns, gpu_alloc_empty(*top.shape), desc)
d_alpha = grad_not_implemented(self, 4, alpha)
d_beta = grad_not_implemented(self, 5, beta)
......@@ -911,7 +911,7 @@ class GpuDnnConv3dGradI(GpuDnnConvGradI):
img = gpu_contiguous(img)
d_kerns = GpuDnnConvGrad3dW()(img, top, gpu_alloc_empty(*kerns.shape), desc)
d_kerns = GpuDnnConv3dGradW()(img, top, gpu_alloc_empty(*kerns.shape), desc)
d_top = GpuDnnConv3d()(img, kerns, gpu_alloc_empty(*top.shape), desc)
d_alpha = grad_not_implemented(self, 4, alpha)
d_beta = grad_not_implemented(self, 5, beta)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论