提交 c49c95fe authored 作者: Mohammed Affan's avatar Mohammed Affan

add staticmethod for cudnn functions

上级 d0930977
...@@ -2158,9 +2158,9 @@ def dconv2di(border_mode, subsample, filter_dilation, num_groups): ...@@ -2158,9 +2158,9 @@ def dconv2di(border_mode, subsample, filter_dilation, num_groups):
class Cudnn_grouped_conv(Grouped_conv_noOptim): class Cudnn_grouped_conv(Grouped_conv_noOptim):
mode = mode_with_gpu mode = mode_with_gpu
conv2d = dconv2d conv2d = staticmethod(dconv2d)
conv2d_gradw = dconv2dw conv2d_gradw = staticmethod(dconv2dw)
conv2d_gradi = dconv2di conv2d_gradi = staticmethod(dconv2di)
conv2d_op = dnn.GpuDnnConv conv2d_op = dnn.GpuDnnConv
conv2d_gradw_op = dnn.GpuDnnConvGradW conv2d_gradw_op = dnn.GpuDnnConvGradW
conv2d_gradi_op = dnn.GpuDnnConvGradI conv2d_gradi_op = dnn.GpuDnnConvGradI
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论