提交 13bad511 authored 作者: carriepl's avatar carriepl 提交者: Frederic

Deprecate algo_bwd flag in favor of algo_bwd_data and algo_bwd_filter

上级 fc7d7277
...@@ -229,16 +229,32 @@ AddConfigVar('dnn.conv.workmem_bwd', ...@@ -229,16 +229,32 @@ AddConfigVar('dnn.conv.workmem_bwd',
EnumStr(''), EnumStr(''),
in_c_key=False) in_c_key=False)
AddConfigVar('dnn.conv.algo_bwd',
"This flag is deprecated; use dnn.conv.algo_bwd_data and "
"dnn.conv.algo_bwd_filter.",
EnumStr(''),
in_c_key=False)
AddConfigVar('dnn.conv.algo_fwd', AddConfigVar('dnn.conv.algo_fwd',
"Default implementation to use for CuDNN forward convolution.", "Default implementation to use for CuDNN forward convolution.",
EnumStr('small', 'none', 'large', 'fft', 'guess_once', EnumStr('small', 'none', 'large', 'fft', 'fft_tiling',
'guess_on_shape_change', 'time_once', 'guess_once', 'guess_on_shape_change',
'time_once', 'time_on_shape_change'),
in_c_key=False)
AddConfigVar('dnn.conv.algo_bwd_data',
"Default implementation to use for CuDNN backward convolution to "
"get the gradients of the convolution with regard to the inputs.",
EnumStr('none', 'deterministic', 'fft', 'fft_tiling',
'guess_once', 'guess_on_shape_change', 'time_once',
'time_on_shape_change'), 'time_on_shape_change'),
in_c_key=False) in_c_key=False)
AddConfigVar('dnn.conv.algo_bwd', AddConfigVar('dnn.conv.algo_bwd_filter',
"Default implementation to use for CuDNN backward convolution.", "Default implementation to use for CuDNN backward convolution to "
EnumStr('none', 'deterministic', 'fft', 'guess_once', "get the gradients of the convolution with regard to the "
"filters.",
EnumStr('none', 'deterministic', 'fft', 'small', 'guess_once',
'guess_on_shape_change', 'time_once', 'guess_on_shape_change', 'time_once',
'time_on_shape_change'), 'time_on_shape_change'),
in_c_key=False) in_c_key=False)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论