like20
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!781 Adjust the order of cast and reshape in the grammar implementation process 

Merge pull request !781 from candanzg/repaire_cast_type
This commit is contained in:
mindspore-ci-bot 2020-04-28 14:37:22 +08:00 committed by Gitee
parent 9ca2349c81 e8850e485e
commit 928b0bb309
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/ops/composite/multitype_ops/setitem_impl.py
View File

@ -247,10 +247,10 @@ def _tensor_assgin_tensor(data, input_slice, value):
indices = mult_util.slice2indices(input_slice, data_shape)
indices_size = F.size(indices)
indices_size = mult_util.check_indices(indices_size, input_slice)
update = F.fill(data_dtype, (indices_size,), 1)
update = F.fill(mstype.int32, (indices_size,), 1)
condition_1d = F.scatter_nd(indices, update, (data_size,))
condition_1d = F.cast(condition_1d, mstype.bool_)
condition = F.reshape(condition_1d, data_shape)
condition = F.cast(condition, mstype.bool_)
# 2. u
value_fill = None
value_size = F.size(value)
@ -325,10 +325,10 @@ def _tensor_assgin_number(data, input_slice, value):
indices = mult_util.slice2indices(input_slice, data_shape)
indices_size = F.size(indices)
indices_size = mult_util.check_indices(indices_size, input_slice)
update = F.fill(data_dtype, (indices_size,), 1)
update = F.fill(mstype.int32, (indices_size,), 1)
condition_1d = F.scatter_nd(indices, update, (data_size,))
condition_1d = F.cast(condition_1d, mstype.bool_)
condition = F.reshape(condition_1d, data_shape)
condition = F.cast(condition, mstype.bool_)
# 2. u
value_fill = F.fill(data_dtype, (indices_size,), value)
value_1d = F.scatter_nd(indices, value_fill, (data_size,))