提交 44f09416 authored 作者: Reyhane Askari's avatar Reyhane Askari

flake8

上级 d0d64720
...@@ -1137,8 +1137,7 @@ def dnn_gradweight(img, topgrad, kerns_shp, border_mode='valid', ...@@ -1137,8 +1137,7 @@ def dnn_gradweight(img, topgrad, kerns_shp, border_mode='valid',
precision = get_precision(precision, [img, topgrad]) precision = get_precision(precision, [img, topgrad])
desc = GpuDnnConvDesc(border_mode=border_mode, subsample=subsample, desc = GpuDnnConvDesc(border_mode=border_mode, subsample=subsample,
conv_mode=conv_mode, precision=precision)( conv_mode=conv_mode, precision=precision)(kerns_shp)
kerns_shp)
out = GpuAllocEmpty(ctx_name, dtype=img.dtype)(*kerns_shp) out = GpuAllocEmpty(ctx_name, dtype=img.dtype)(*kerns_shp)
return GpuDnnConvGradW()(img, topgrad, out, desc) return GpuDnnConvGradW()(img, topgrad, out, desc)
...@@ -1151,6 +1150,7 @@ def dnn_gradweight3d(img, topgrad, kerns_shp, border_mode='valid', ...@@ -1151,6 +1150,7 @@ def dnn_gradweight3d(img, topgrad, kerns_shp, border_mode='valid',
return dnn_gradweight(img, topgrad, kerns_shp, border_mode, return dnn_gradweight(img, topgrad, kerns_shp, border_mode,
subsample, conv_mode, precision) subsample, conv_mode, precision)
def dnn_gradinput(kerns, topgrad, img_shp, border_mode='valid', def dnn_gradinput(kerns, topgrad, img_shp, border_mode='valid',
subsample=(1, 1), conv_mode='conv', precision=None): subsample=(1, 1), conv_mode='conv', precision=None):
""" """
...@@ -1165,8 +1165,7 @@ def dnn_gradinput(kerns, topgrad, img_shp, border_mode='valid', ...@@ -1165,8 +1165,7 @@ def dnn_gradinput(kerns, topgrad, img_shp, border_mode='valid',
precision = get_precision(precision, [kerns, topgrad]) precision = get_precision(precision, [kerns, topgrad])
desc = GpuDnnConvDesc(border_mode=border_mode, subsample=subsample, desc = GpuDnnConvDesc(border_mode=border_mode, subsample=subsample,
conv_mode=conv_mode, precision=precision)( conv_mode=conv_mode, precision=precision)(kerns.shape)
kerns.shape)
out = GpuAllocEmpty(ctx_name, kerns.dtype)(*img_shp) out = GpuAllocEmpty(ctx_name, kerns.dtype)(*img_shp)
return GpuDnnConvGradI()(kerns, topgrad, out, desc) return GpuDnnConvGradI()(kerns, topgrad, out, desc)
...@@ -1179,6 +1178,7 @@ def dnn_gradinput3d(kerns, topgrad, img_shp, border_mode='valid', ...@@ -1179,6 +1178,7 @@ def dnn_gradinput3d(kerns, topgrad, img_shp, border_mode='valid',
return dnn_gradinput(kerns, topgrad, img_shp, border_mode, subsample, return dnn_gradinput(kerns, topgrad, img_shp, border_mode, subsample,
conv_mode, precision) conv_mode, precision)
class GpuDnnPoolDesc(Op): class GpuDnnPoolDesc(Op):
""" """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论