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

Move ShapeOpt after first merge

上级 81ccd9d7
...@@ -1250,9 +1250,10 @@ class ShapeOptimizer(Optimizer): ...@@ -1250,9 +1250,10 @@ class ShapeOptimizer(Optimizer):
def apply(self, fgraph): def apply(self, fgraph):
pass pass
# -1 should make it run right before the first merge # Register it after merge1 optimization at 0. We don't want to track
# the shape of merged node.
theano.compile.mode.optdb.register('ShapeOpt', ShapeOptimizer(), theano.compile.mode.optdb.register('ShapeOpt', ShapeOptimizer(),
-1, 'fast_run', 'fast_compile') 0.1, 'fast_run', 'fast_compile')
@register_specialize @register_specialize
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论