huawei
/
mindspore2022
0
0
Fork 53
Code Issues Pull Requests 29 Projects Releases Wiki Activity

update output size of reshape op when dynamic shape

This commit is contained in:
zhengzuohe 2022-04-06 17:28:19 +08:00
parent e611ca2907
commit 738f019940
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/plugin/device/ascend/kernel/host/host_kernel_mod.cc
View File

@ -81,6 +81,10 @@ void HostKernelMod::InferOp() {
MS_LOG(EXCEPTION) << "The node is not dynamic shape.";
}
KernelMod::InferShape();
input_size_list_.clear();
output_size_list_.clear();
HostKernelMod::Init(node);
}
std::vector<TaskInfoPtr> HostKernelMod::GenTask(const std::vector<AddressPtr> &, const std::vector<AddressPtr> &,