提交 fa8e31d6 authored 作者: Frederic Bastien's avatar Frederic Bastien

Revert part of "Alloc OpRemove to accept a class to remove. Tested by…

Revert part of "Alloc OpRemove to accept a class to remove. Tested by test_gradien.py:test_grad_clip" This reverts part of commit 0e2d7a79. We don't need that and it is safer to not introduce it.
上级 52ce429f
......@@ -1545,10 +1545,8 @@ class OpRemove(LocalOptimizer):
return [self.op]
def transform(self, node):
if inspect.isclass(self.op):
if isinstance(node.op, self.op):
return node.inputs
elif node.op == self.op:
if node.op != self.op:
return False
return node.inputs
def __str__(self):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论