p12346870
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!21413 Fix "Dynamic shape execution failed with an error cast failed" 

Merge pull request !21413 from zhangzhaoju/master_dynamic_shape_cast_failure
This commit is contained in:
i-robot 2021-08-06 01:26:59 +00:00 committed by Gitee
parent 4583f298b0 52779eabb5
commit 054abd080b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ops/operations/array_ops.py
View File

@ -1203,7 +1203,7 @@ class Size(PrimitiveWithInfer):
else:
size = functools.reduce(lambda x, y: x * y, x['shape'])
out = {'shape': None,
'dtype': mstype.int32,
'dtype': mstype.int64,
'value': size}
return out