提交 8e089e7a authored 作者: Frederic's avatar Frederic

Register constant_folding during gpu opt.

This give a small speed up.
上级 e236a056
......@@ -83,6 +83,8 @@ def register_opt(*tags, **kwargs):
#register local_track_shape_i at this level too
#to make multi-level lift of shape work.
register_opt()(theano.tensor.opt.local_track_shape_i)
register_opt(name='gpu_constant_folding')(
tensor.opt.constant_folding)
class InputToGpuOptimizer(Optimizer):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论