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

!46173 fix an issue of dynamic obfuscation 

Merge pull request !46173 from jxl/master
This commit is contained in:
i-robot 2022-11-29 09:18:11 +00:00 committed by Gitee
parent c24d7910f6 5ef0a90a44
commit 6f7a955f4b
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/python/mindspore/nn/cell.py
View File

@ -2244,6 +2244,7 @@ class GraphCell(Cell):
raise ValueError(
"'obf_password' must be larger than 0, and less or equal than int64 ({}),"
"but got {}.".format(int_64_max, obf_password))
self._obf_password, self._append_password = _generate_pair_password(self.obf_password)
params_init = {} if params_init is None else params_init
if not isinstance(params_init, dict):
raise TypeError(f"For 'GraphCell', the argument 'params_init' must be a dict, but got {type(params_init)}.")
@ -2265,9 +2266,8 @@ class GraphCell(Cell):
self._add_attr("graph_load_from_mindir", self.graph)
if not self.obf_password:
return self.compile_and_run(*inputs)
obf_password, append_password = _generate_pair_password(self.obf_password)
append_input_1 = Tensor((numpy.ones((1, 1)) * obf_password).astype(numpy.int32))
append_input_2 = Tensor((numpy.ones((1, 1)) * append_password).astype(numpy.int32))
append_input_1 = Tensor((numpy.ones((1, 1)) * self._obf_password).astype(numpy.int32))
append_input_2 = Tensor((numpy.ones((1, 1)) * self._append_password).astype(numpy.int32))
return self.compile_and_run(*inputs, append_input_1, append_input_2)