提交 627d9bf7 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix bad doc syntax in cuda/dnn.

上级 81841795
...@@ -579,9 +579,9 @@ class GpuDnnConv3d(GpuDnnConv): ...@@ -579,9 +579,9 @@ class GpuDnnConv3d(GpuDnnConv):
:param descr: the convolution descriptor :param descr: the convolution descriptor
:param workmem: :param workmem:
*deprecated*, use parameter algo instead. *deprecated*, use parameter algo instead.
:param algo: ['none', 'guess_once', 'guess_on_shape_change', :param algo: ['none', 'guess_once', 'guess_on_shape_change', 'time_once', 'time_on_shape_change']
'time_once', 'time_on_shape_change'] Default is the value of :attr:`config.dnn.conv.algo_fwd`.
Default is the value of :attr:`config.dnn.conv.algo_fwd.
""" """
__props__ = ('algo', 'inplace') __props__ = ('algo', 'inplace')
__input_name__ = ('image', 'kernel', 'output', __input_name__ = ('image', 'kernel', 'output',
...@@ -683,11 +683,7 @@ class GpuDnnConvGradW(DnnBase, COp): ...@@ -683,11 +683,7 @@ class GpuDnnConvGradW(DnnBase, COp):
The convolution descriptor. The convolution descriptor.
workmem workmem
*deprecated*, use parameter algo instead. *deprecated*, use parameter algo instead.
algo algo : {'none', 'deterministic', 'fft', 'guess_once', 'guess_on_shape_change', 'time_once', 'time_on_shape_change'}
['none', 'deterministic', 'fft', 'guess_once',
'guess_on_shape_change', 'time_once',
'time_on_shape_change']
Default is the value of :attr:`config.dnn.conv.algo_bwd`. Default is the value of :attr:`config.dnn.conv.algo_bwd`.
""" """
...@@ -822,8 +818,7 @@ class GpuDnnConv3dGradW(GpuDnnConvGradW): ...@@ -822,8 +818,7 @@ class GpuDnnConv3dGradW(GpuDnnConvGradW):
:param descr: the convolution descriptor :param descr: the convolution descriptor
:param workmem: :param workmem:
*deprecated*, use parameter algo instead. *deprecated*, use parameter algo instead.
:param algo: ['none', 'guess_once', 'guess_on_shape_change', :param algo: ['none', 'guess_once', 'guess_on_shape_change', 'time_once', 'time_on_shape_change']
'time_once', 'time_on_shape_change']
Default is the value of :attr:`config.dnn.conv.algo_bwd`. Default is the value of :attr:`config.dnn.conv.algo_bwd`.
""" """
...@@ -1095,8 +1090,8 @@ def dnn_conv(img, kerns, border_mode='valid', subsample=(1, 1), ...@@ -1095,8 +1090,8 @@ def dnn_conv(img, kerns, border_mode='valid', subsample=(1, 1),
""" """
GPU convolution using cuDNN from NVIDIA. GPU convolution using cuDNN from NVIDIA.
The memory layout to use is 'bc01', that is 'batch', 'channel', 'first dim', The memory layout to use is 'bc01', that is 'batch', 'channel',
'second dim' in that order. 'first dim', 'second dim' in that order.
Parameters Parameters
---------- ----------
...@@ -1129,7 +1124,7 @@ def dnn_conv(img, kerns, border_mode='valid', subsample=(1, 1), ...@@ -1129,7 +1124,7 @@ def dnn_conv(img, kerns, border_mode='valid', subsample=(1, 1),
algo : {'none', 'small', 'large', 'fft', 'guess_once', 'guess_on_shape_change', 'time_once', 'time_on_shape_change'} algo : {'none', 'small', 'large', 'fft', 'guess_once', 'guess_on_shape_change', 'time_once', 'time_on_shape_change'}
Convolution implementation to use. Some of its values may require certain Convolution implementation to use. Some of its values may require certain
versions of CuDNN to be installed. Default is the value of versions of CuDNN to be installed. Default is the value of
:attr:`config.dnn.conv.algo_fwd. :attr:`config.dnn.conv.algo_fwd`.
""" """
...@@ -1221,7 +1216,7 @@ def dnn_conv3d(img, kerns, border_mode='valid', subsample=(1, 1, 1), ...@@ -1221,7 +1216,7 @@ def dnn_conv3d(img, kerns, border_mode='valid', subsample=(1, 1, 1),
:param workmem: *deprecated*, use param algo instead :param workmem: *deprecated*, use param algo instead
:param algo: convolution implementation to use. Only 'none' is implemented :param algo: convolution implementation to use. Only 'none' is implemented
for the conv3d. Default is the value of for the conv3d. Default is the value of
:attr:`config.dnn.conv.algo_fwd. :attr:`config.dnn.conv.algo_fwd`.
:warning: The cuDNN library only works with GPU that have a compute :warning: The cuDNN library only works with GPU that have a compute
capability of 3.0 or higer. This means that older GPU will not capability of 3.0 or higer. This means that older GPU will not
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论