提交 67812dbc authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix broken condition checking for float16 in Composite.

上级 1c0cf112
......@@ -3308,8 +3308,8 @@ class Composite(ScalarOp):
raise ValueError(
"All orphans in the fgraph to Composite must"
" be Constant instances.")
elif any(i.dtype == 'float16' for i in var.owner.inputs or
o.dtype == 'float16' for o in var.owner.outputs):
elif (any(i.dtype == 'float16' for i in var.owner.inputs) or
any(o.dtype == 'float16' for o in var.owner.outputs):
# flag for elemwise ops to check.
self.inner_float16 = True
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论