提交 e46d2366 authored 作者: Mathieu Germain's avatar Mathieu Germain

Updated docstring in new backend

上级 dc2277cc
...@@ -402,10 +402,10 @@ class GpuDnnConv(DnnBase): ...@@ -402,10 +402,10 @@ class GpuDnnConv(DnnBase):
kernel kernel
descr descr
The convolution descriptor. The convolution descriptor.
algo : {'small', 'none', 'large', 'fft', 'fft_tiling', 'winograd', 'guess_once', algo
'guess_on_shape_change', 'time_once', 'time_on_shape_change'} {'small', 'none', 'large', 'fft', 'fft_tiling', 'winograd', 'guess_once',
'guess_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')
...@@ -577,7 +577,10 @@ class GpuDnnConvGradW(DnnBase): ...@@ -577,7 +577,10 @@ class GpuDnnConvGradW(DnnBase):
kernel kernel
descr descr
The convolution descriptor. The convolution descriptor.
algo
{'none', 'deterministic', 'fft', 'small', 'guess_once',
'guess_on_shape_change', 'time_once', 'time_on_shape_change'}
Default is the value of :attr:`config.dnn.conv.algo_bwd_filter`.
""" """
__props__ = ('algo', 'inplace') __props__ = ('algo', 'inplace')
...@@ -696,7 +699,10 @@ class GpuDnnConvGradI(DnnBase): ...@@ -696,7 +699,10 @@ class GpuDnnConvGradI(DnnBase):
kernel kernel
descr descr
The convolution descriptor. The convolution descriptor.
algo
{'none', 'deterministic', 'fft', 'fft_tiling', 'winograd', 'guess_once',
'guess_on_shape_change', 'time_once', 'time_on_shape_change'}
Default is the value of :attr:`config.dnn.conv.algo_bwd_data`.
""" """
__props__ = ('algo', 'inplace',) __props__ = ('algo', 'inplace',)
...@@ -716,9 +722,12 @@ class GpuDnnConvGradI(DnnBase): ...@@ -716,9 +722,12 @@ class GpuDnnConvGradI(DnnBase):
if version() < 4000 and self.algo == 'fft_tiling': if version() < 4000 and self.algo == 'fft_tiling':
raise RuntimeError("CuDNN's tiled-FFT convolution requires CuDNN " raise RuntimeError("CuDNN's tiled-FFT convolution requires CuDNN "
"v4 or more recent") "v4 or more recent")
if version() < 5000 and self.algo == 'winograd':
raise RuntimeError("CuDNN's winograd convolution requires CuDNN "
"v5 or more recent")
assert self.algo in ['none', 'deterministic', 'fft', 'fft_tiling', assert self.algo in ['none', 'deterministic', 'fft', 'fft_tiling',
'guess_once', 'guess_on_shape_change', 'winograd', 'guess_once', 'guess_on_shape_change',
'time_once', 'time_on_shape_change'] 'time_once', 'time_on_shape_change']
def __setstate__(self, d): def __setstate__(self, d):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论