提交 67ca51b6 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Add MergeOptimizer as a part of canonicalize

This makes it easier to do some simplifications.
上级 5d226894
......@@ -31,6 +31,7 @@ from theano import compile # to register the optimizer built by this file
from theano.gof.python25 import any, all
from theano.gof.opt import (Optimizer, pre_constant_merge,
pre_greedy_local_optimizer)
from theano.gof.opt import merge_optimizer
from theano.gof import toolbox, DestroyHandler
from basic import get_constant_value, ShapeError
......@@ -301,6 +302,11 @@ def register_specialize_device(lopt, *tags, **kwargs):
return lopt
## Register merge_optimizer as a global opt during canonicalize
compile.optdb['canonicalize'].register(
'canon_merge', merge_optimizer, 'fast_run')
#####################
# Dot optimizations #
#####################
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论