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

register local opt in gpu opt to clean up the graph

上级 91e498ae
...@@ -87,6 +87,7 @@ gpu_optimizer.register('gpu_merge', theano.gof.opt.merge_optimizer, ...@@ -87,6 +87,7 @@ gpu_optimizer.register('gpu_merge', theano.gof.opt.merge_optimizer,
register_opt()(theano.tensor.opt.local_track_shape_i) register_opt()(theano.tensor.opt.local_track_shape_i)
register_opt(name='gpu_constant_folding')( register_opt(name='gpu_constant_folding')(
tensor.opt.constant_folding) tensor.opt.constant_folding)
register_opt()(theano.tensor.opt.local_subtensor_make_vector)
# This is a partial list of CPU ops that can be in some circonstance # This is a partial list of CPU ops that can be in some circonstance
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论