提交 dc29b28f authored 作者: Frederic Bastien's avatar Frederic Bastien

register local_track_shape_i into the gpu optimization to make all the lift their too.

上级 6f9adf92
......@@ -29,6 +29,9 @@ def register_opt(*tags, **kwargs):
return local_opt
return f
#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)
class InputToGpuOptimizer(Optimizer):
"""Transfert the input of a graph to the gpu if needed
It should make this part of the optimizer faster we will will need only 1 pass on the env.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论