提交 1f39a9f1 编写于 作者: S ShenLiang 提交者: Yi Liu

fix the bug of scatter_nd, test=develop (#21257)

上级 382cf5d7
......@@ -7510,7 +7510,7 @@ def scatter_nd_add(ref, index, updates, name=None):
raise ValueError("ref and updates must have same data type.")
helper = LayerHelper('scatter_nd_add', **locals())
dtype = helper.input_dtype()
dtype = helper.input_dtype(input_param_name='ref')
if name is None:
output = helper.create_variable_for_type_inference(dtype)
else:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册