提交 76e8a8ab authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Forward test value in clone_float32

上级 f62366bc
...@@ -4078,6 +4078,8 @@ class Compositef32(object): ...@@ -4078,6 +4078,8 @@ class Compositef32(object):
for i in fgraph.inputs: for i in fgraph.inputs:
if i.dtype == 'float16': if i.dtype == 'float16':
mapping[i] = get_scalar_type('float32')() mapping[i] = get_scalar_type('float32')()
if hasattr(i.tag, 'test_value'):
mapping[i].tag.test_value = i.tag.test_value
else: else:
mapping[i] = i mapping[i] = i
for node in topo: for node in topo:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论