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

Fix broken condition checking for float16 in Composite.

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