Skip to content

P
pytensor
提交 b424cc1d authored 作者: Mathieu Germain's avatar Mathieu Germain
浏览文件
操作

updated comment

上级 dab7a1ed
隐藏空白字符变更
内嵌 并排
正在显示 包含 1 行增加 和 1 行删除
theano/sandbox/cuda/dnn.py
浏览文件 @ b424cc1d
...@@ -390,7 +390,7 @@ class GpuDnnConv(DnnBase, COp): ...@@ -390,7 +390,7 @@ class GpuDnnConv(DnnBase, COp):
# need v4 for conv2d, need v5 for conv3d # need v4 for conv2d, need v5 for conv3d
alg = 'CUDNN_CONVOLUTION_FWD_ALGO_FFT_TILING' alg = 'CUDNN_CONVOLUTION_FWD_ALGO_FFT_TILING'
elif self.algo == 'winograd': elif self.algo == 'winograd':
# need v4 for conv2d, need v5 for conv3d # need v5
alg = 'CUDNN_CONVOLUTION_FWD_ALGO_WINOGRAD' alg = 'CUDNN_CONVOLUTION_FWD_ALGO_WINOGRAD'
elif self.algo in ['guess_once', 'guess_on_shape_change']: elif self.algo in ['guess_once', 'guess_on_shape_change']:
# The convolution implementation should be choosen according # The convolution implementation should be choosen according
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论