提交 41ee67e9 authored 作者: Amjad Almahairi's avatar Amjad Almahairi

registering local optimization for Solve op

上级 bf1d0950
......@@ -286,7 +286,7 @@ def inv_as_solve(node):
@register_stabilize
@register_canonicalize
@local_optimizer(None) # XXX: solve is defined later and can't be used here
@local_optimizer([Solve])
def tag_solve_triangular(node):
"""
If a general solve() is applied to the output of a cholesky op, then
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论