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

!59782 fix bug in lite 

Merge pull request !59782 from XinDu/master
This commit is contained in:
i-robot 2023-09-26 13:30:46 +00:00 committed by Gitee
parent fe564e3a31 66b2d8d5d5
commit be34fa9f07
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/lite/tools/kernel_builder/ascend/akg/custom.py
View File

@ -432,10 +432,8 @@ class Reshape(OpInfer):
self.output_desc[0]["ori_shape"] = out_shape
def post_process(self):
for item in self.op_desc["attr"]:
if item["name"] == "shape":
item["ori_value"] = item["value"]
item["value"] = self.output_desc[0]["shape"]
self.input_desc[1]["ori_value"] = self.input_desc[1]["value"]
self.input_desc[1]["value"] = self.output_desc[0]["shape"]
class ExpandDimAndSqueeze(Reshape):
def copy_axis(self, axis):