提交 d6ad187a authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

flake8

上级 1da5a694
......@@ -3838,7 +3838,7 @@ class Compositef32(object):
self.special[type(node.op)](node, mapping)
continue
new_node = node.clone_with_new_inputs(
[mapping[i] for i in node.inputs],
[mapping[inp] for inp in node.inputs],
strict=False)
# make sure we don't produce any float16.
assert not any(o.dtype == 'float16' for o in new_node.outputs)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论