提交 15da8ef6 authored 作者: --global's avatar --global

Add params to feature NoOutputFromInplace

上级 47442dc1
...@@ -440,10 +440,21 @@ class PreserveNames(Feature): ...@@ -440,10 +440,21 @@ class PreserveNames(Feature):
class NoOutputFromInplace(Feature): class NoOutputFromInplace(Feature):
def __init__(self, first_output_idx=0, last_output_idx=None):
self.first_idx = first_output_idx
self.last_idx = last_output_idx
def validate(self, fgraph): def validate(self, fgraph):
if not hasattr(fgraph, 'destroyers'): if not hasattr(fgraph, 'destroyers'):
return True return True
for out in list(fgraph.outputs):
if self.last_idx is None:
outputs_to_validate = list(fgraph.outputs)[self.first_idx:]
else:
outputs_to_validate = list(fgraph.outputs)[self.first_idx:
self.last_idx]
for out in outputs_to_validate:
if out.owner is None: if out.owner is None:
continue continue
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论