mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

fix GetInputs mem leak

This commit is contained in:
yefeng 2022-08-15 13:58:05 +08:00
parent 984a92c0b8
commit ba7f43cabe
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/lite/src/extendrt/cxx_api/model_pool/model_pool.cc
View File

@ -652,8 +652,8 @@ std::vector<MSTensor> ModelPool::GetInputs() {
return {};
}
for (size_t i = 0; i < model_pool_inputs_.size(); i++) {
auto tensor = mindspore::MSTensor::CreateRefTensor(model_pool_inputs_.at(i).Name(),
model_pool_inputs_.at(i).DataType(), {}, nullptr, 0);
auto tensor = mindspore::MSTensor::CreateTensor(model_pool_inputs_.at(i).Name(),
model_pool_inputs_.at(i).DataType(), {}, nullptr, 0);
if (tensor == nullptr) {
MS_LOG(ERROR) << "create tensor failed.";
return {};
@ -675,8 +675,8 @@ std::vector<MSTensor> ModelPool::GetOutputs() {
return {};
}
for (size_t i = 0; i < model_pool_outputs_.size(); i++) {
auto tensor = mindspore::MSTensor::CreateRefTensor(model_pool_outputs_.at(i).Name(),
model_pool_outputs_.at(i).DataType(), {}, nullptr, 0);
auto tensor = mindspore::MSTensor::CreateTensor(model_pool_outputs_.at(i).Name(),
model_pool_outputs_.at(i).DataType(), {}, nullptr, 0);
if (tensor == nullptr) {
MS_LOG(ERROR) << "create tensor failed.";
return {};