fix the bug that input tensor data with old size dosen't free
This commit is contained in:
parent
9daf5933e1
commit
ea6368c98c
|
@ -158,6 +158,7 @@ int LiteOpActor::ResizeGraphInput(const std::vector<mindspore::tensor::MSTensor
|
|||
auto src_tensor = map.second;
|
||||
for (size_t i = 0; i < inputs.size(); i++) {
|
||||
if (src_tensor == inputs[i]) {
|
||||
isolate_tensor->FreeData();
|
||||
isolate_tensor->set_shape(dims[i]);
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue