提交 19619e9a authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #4549 from slefrancois/update_doc_conv

Update doc conv flags
...@@ -641,35 +641,54 @@ import theano and print the config variable, as in: ...@@ -641,35 +641,54 @@ import theano and print the config variable, as in:
.. attribute:: config.dnn.conv.workmem .. attribute:: config.dnn.conv.workmem
Deprecated, use dnn.conv.algo_fwd. Deprecated, use :attr:`config.dnn.conv.algo_fwd`.
.. attribute:: config.dnn.conv.workmem_bwd .. attribute:: config.dnn.conv.workmem_bwd
Deprecated, use dnn.conv.algo_bwd. Deprecated, use :attr:`config.dnn.conv.algo_bwd_filter` and
:attr:`config.dnn.conv.algo_bwd_data` instead.
.. attribute:: config.dnn.conv.algo_fwd .. attribute:: config.dnn.conv.algo_fwd
String value: ``small``, ``none``, ``large``, ``fft``, ``guess_once``, String value:
``guess_on_shape_change``, ``time_once``, ``small``, ``none``, ``large``, ``fft``, ``fft_tiling``,
``time_on_shape_change``. ``winograd``, ``guess_once``, ``guess_on_shape_change``,
``time_once``, ``time_on_shape_change``.
Default: ``small`` Default: ``small``
3d convolution only support ``none``, ``guess_once``,
``guess_on_shape_change``, ``time_once``, ``time_on_shape_change``.
3d convolution only support ``none``, ``winograd``, ``guess_once``,
``guess_on_shape_change``, ``time_once``, ``time_on_shape_change``.
.. attribute:: config.dnn.conv.algo_bwd .. attribute:: config.dnn.conv.algo_bwd
String value: ``none``, ``deterministic``, ``fft``, ``guess_once``, Deprecated, use :attr:`config.dnn.conv.algo_bwd_filter` and
``guess_on_shape_change``, ``time_once``, :attr:`config.dnn.conv.algo_bwd_data` instead.
``time_on_shape_change``.
.. attribute:: config.dnn.conv.algo_bwd_filter
String value:
``none``, ``deterministic``, ``fft``, ``small``, ``guess_once``,
``guess_on_shape_change``, ``time_once``, ``time_on_shape_change``.
Default: ``none``
3d convolution only supports ``none``, ``guess_once``,
``guess_on_shape_change``, ``time_once``, ``time_on_shape_change``.
.. attribute:: config.dnn.conv.algo_bwd_data
String value:
``none``, ``deterministic``, ``fft``, ``fft_tiling``, ``winograd``,
``guess_once``, ``guess_on_shape_change``, ``time_once``,
``time_on_shape_change``.
Default: ``none`` Default: ``none``
3d convolution only support ``none``, ``guess_once``, 3d convolution only support ``none``, ``winograd``,
``guess_on_shape_change``, ``time_once``, ``time_on_shape_change``. ``guess_once``, ``guess_on_shape_change``, ``time_once``,
``time_on_shape_change``.
.. attribute:: config.gcc.cxxflags .. attribute:: config.gcc.cxxflags
......
...@@ -273,7 +273,8 @@ def safe_no_dnn_workmem_bwd(workmem): ...@@ -273,7 +273,8 @@ def safe_no_dnn_workmem_bwd(workmem):
return True return True
AddConfigVar('dnn.conv.workmem_bwd', AddConfigVar('dnn.conv.workmem_bwd',
"This flag is deprecated; use dnn.conv.algo_bwd.", "This flag is deprecated; use `dnn.conv.algo_bwd_filter` "
"and `dnn.conv.algo_bwd_data` instead.",
ConfigParam('', allow_override=False, ConfigParam('', allow_override=False,
filter=safe_no_dnn_workmem_bwd), filter=safe_no_dnn_workmem_bwd),
in_c_key=False) in_c_key=False)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论