提交 755429dd authored 作者: --global's avatar --global

Update comment in GpuDnnConv

上级 2597dcd2
...@@ -386,8 +386,8 @@ class GpuDnnConv(DnnBase, COp): ...@@ -386,8 +386,8 @@ class GpuDnnConv(DnnBase, COp):
def __init__(self, workmem=None, inplace=False): def __init__(self, workmem=None, inplace=False):
""" """
:param workmem: either 'none', 'small' or 'large'. Default is :param workmem: either 'none', 'small', 'large', 'fft', 'time' or
the value of :attr:`config.dnn.conv.workmem`. 'guess'. Default is the value of :attr:`config.dnn.conv.workmem`.
""" """
COp.__init__(self, ["dnn_base.c", "dnn_conv_base.c", "dnn_fwd.c"], COp.__init__(self, ["dnn_base.c", "dnn_conv_base.c", "dnn_fwd.c"],
"APPLY_SPECIFIC(conv_fwd)") "APPLY_SPECIFIC(conv_fwd)")
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论