提交 0ae495e1 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix logic of the float16 check.

上级 dd98172a
...@@ -3301,14 +3301,15 @@ class Composite(ScalarOp): ...@@ -3301,14 +3301,15 @@ class Composite(ScalarOp):
["%%(o%i)s" % i for i in xrange(len(self.fgraph.outputs))])) ["%%(o%i)s" % i for i in xrange(len(self.fgraph.outputs))]))
for var in self.fgraph.variables: for var in self.fgraph.variables:
if var.owner is None and var not in self.fgraph.inputs: if var.owner is None:
# This is an orphan if var not in self.fgraph.inputs:
if isinstance(var, Constant): # This is an orphan
subd[var] = var.type.c_literal(var.data) if isinstance(var, Constant):
else: subd[var] = var.type.c_literal(var.data)
raise ValueError( else:
"All orphans in the fgraph to Composite must" raise ValueError(
" be Constant instances.") "All orphans in the fgraph to Composite must"
" 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): o.dtype == 'float16' for o in var.owner.outputs):
# flag for elemwise ops to check. # flag for elemwise ops to check.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论