提交 3914bf65 authored 作者: carriepl's avatar carriepl 提交者: Frederic

Add new flag to control CuDNN convolution precision

上级 c36f9a29
......@@ -259,6 +259,12 @@ AddConfigVar('dnn.conv.algo_bwd_filter',
'time_on_shape_change'),
in_c_key=False)
AddConfigVar('dnn.conv.precision',
"Default data precision to use for the computation in CuDNN "
"convolutions (defaults to the floatX).",
EnumStr('floatX', 'float16', 'float32', 'float64'),
in_c_key=False)
def default_dnn_path(suffix):
def f(suffix=suffix):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论