提交 71b70e33 authored 作者: James Bergstra's avatar James Bergstra

Added register_stabilize() to tensor.opt

上级 5165df44
...@@ -140,6 +140,11 @@ def register_specialize(lopt, *tags, **kwargs): ...@@ -140,6 +140,11 @@ def register_specialize(lopt, *tags, **kwargs):
compile.optdb['specialize'].register(name, lopt, 'fast_run', *tags) compile.optdb['specialize'].register(name, lopt, 'fast_run', *tags)
return lopt return lopt
def register_stabilize(lopt, *tags, **kwargs):
name = (kwargs and kwargs.pop('name')) or lopt.__name__
compile.optdb['stabilize'].register(name, lopt, 'fast_run', *tags)
return lopt
###################### ######################
# DimShuffle lifters # # DimShuffle lifters #
###################### ######################
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论