提交 cabf5754 authored 作者: Frederic's avatar Frederic

Better printing of a optimizer

上级 46423b3e
......@@ -1109,9 +1109,10 @@ def local_gpu_softmax_with_bias(node):
from theano.tensor.nnet import conv
# We need a fixed order for the user interface.
conv_groupopt = theano.gof.optdb.LocalGroupDB()
conv_groupopt.__name__ = "gpu_conv_opt"
conv_groupopt.__name__ = "gpu_conv_opts"
register_opt('fast_compile', 'fast_run', 'gpu')(conv_groupopt)
def _gpu_conv_to_fftconv(node):
# shared helper function for local_conv_fft_valid and local_conv_fft_full.
# we import conv2d_fft locally to avoid pycuda warnings
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论