提交 f2e2a7be authored 作者: --global's avatar --global

Modify dnn.conv.algo_fwd flag to support timing feature

上级 fc1ed897
...@@ -360,7 +360,8 @@ AddConfigVar('dnn.conv.algo_fwd', ...@@ -360,7 +360,8 @@ AddConfigVar('dnn.conv.algo_fwd',
AddConfigVar('dnn.conv.algo_bwd', AddConfigVar('dnn.conv.algo_bwd',
"Default implementation to use for CuDNN backward convolution.", "Default implementation to use for CuDNN backward convolution.",
EnumStr('none', 'deterministic', 'fft', 'guess_once', EnumStr('none', 'deterministic', 'fft', 'guess_once',
'guess_on_shape_change'), 'guess_on_shape_change', 'time_once',
'time_on_shape_change'),
in_c_key=False) in_c_key=False)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论