提交 19a28b16 编写于 作者: R robot 提交者: Yu Yang

[fix assign op]/2 using assign op instead of scale1.0

上级 8806801e
......@@ -195,7 +195,7 @@ def assign(input, output):
helper.append_op(
type='assign',
inputs={'X': [input]},
outputs={'Out': [output]},
outputs={'Out': [output]})
elif isinstance(input, numpy.ndarray):
dtype = convert_np_dtype_to_dtype_(input.dtype)
if dtype == VarDesc.VarType.FP32:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册