提交 ead48442 authored 作者: --global's avatar --global

Add params to opt AddNoOutputFromInplace

上级 15da8ef6
......@@ -170,9 +170,14 @@ class AddNoOutputFromInplace(gof.Optimizer):
outputs
"""
def __init__(self, first_output_idx=0, last_output_idx=None):
self.feature = gof.NoOutputFromInplace(first_output_idx,
last_output_idx)
def add_requirements(self, fgraph):
super(AddNoOutputFromInplace, self).add_requirements(fgraph)
fgraph.attach_feature(gof.NoOutputFromInplace())
fgraph.attach_feature(self.feature)
class PrintCurrentFunctionGraph(gof.Optimizer):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论