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

intel-update-authors
robot 7 years ago committed by Yu Yang
parent 76c4ae856f
commit 8806801eb8

@ -193,10 +193,9 @@ def assign(input, output):
helper = LayerHelper('assign', **locals())
if isinstance(input, Variable):
helper.append_op(
type='scale',
type='assign',
inputs={'X': [input]},
outputs={'Out': [output]},
attrs={'scale': 1.0})
elif isinstance(input, numpy.ndarray):
dtype = convert_np_dtype_to_dtype_(input.dtype)
if dtype == VarDesc.VarType.FP32:

Loading…
Cancel
Save