xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!47983 fix set inputs problem 

Merge pull request !47983 from luochao60/fix_set_inputs_problem_20230117
This commit is contained in:
i-robot 2023-01-18 09:42:50 +00:00 committed by Gitee
parent fa21a9f933 ce5241dda8
commit f4752a8ab8
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
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/python/mindspore/common/api.py
View File

@ -410,7 +410,7 @@ class _MindsporeFunctionExecutor:
# Case: If the shape of input args is dynamic, get dynamic shape tensor from context and use it to compile.
compile_args = args_list
# Case: The `set_inputs()` of Cell object has been set, using these dynamic shape args as compile args.
if isinstance(self.obj, ms.nn.Cell) and self.obj.get_inputs():
if self.fn.__name__ == 'construct' and isinstance(self.obj, ms.nn.Cell) and self.obj.get_inputs():
compile_args = self.obj.get_inputs()
for args in compile_args:
Validator.check_isinstance("args set in `set_inputs()` of Cell", args, PythonTensor)