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

Tag opt fast_compile to help clean up the graph

上级 ac631700
...@@ -2414,7 +2414,8 @@ def local_useless_subtensor(node): ...@@ -2414,7 +2414,8 @@ def local_useless_subtensor(node):
return [node.inputs[0]] return [node.inputs[0]]
@register_canonicalize # fast_compile to allow opt subtensor(cast{float32}(make_vector))
@register_canonicalize('fast_compile')
@gof.local_optimizer([Subtensor]) @gof.local_optimizer([Subtensor])
def local_subtensor_lift(node): def local_subtensor_lift(node):
""" """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论