diff --git a/mindspore/lite/src/runtime/kernel/arm/base/pooling_base.cc b/mindspore/lite/src/runtime/kernel/arm/base/pooling_base.cc index 7a1cc29c23..f1e9a61da0 100644 --- a/mindspore/lite/src/runtime/kernel/arm/base/pooling_base.cc +++ b/mindspore/lite/src/runtime/kernel/arm/base/pooling_base.cc @@ -48,6 +48,7 @@ int PoolingBaseCPUKernel::SetQuantParam() { free(*pooling_quant_arg_); free(pooling_quant_arg_); pooling_quant_arg_ = nullptr; + *this->pooling_quant_arg_ = nullptr; return RET_MEMORY_FAILED; } auto *input_tensor = in_tensors_.at(kInputIndex); diff --git a/mindspore/lite/tools/anf_importer/import_from_meta_graphT.cc b/mindspore/lite/tools/anf_importer/import_from_meta_graphT.cc index f26c9a7d6a..6f20b140a9 100644 --- a/mindspore/lite/tools/anf_importer/import_from_meta_graphT.cc +++ b/mindspore/lite/tools/anf_importer/import_from_meta_graphT.cc @@ -61,6 +61,7 @@ int AnfImporterFromMetaGraphT::ConverterConstTensor() { auto ret = memcpy_s(tensor_data, size, tensor->data.data(), size); if (EOK != ret) { MS_LOG(ERROR) << "memcpy_s error"; + delete[] tensor_data; return RET_MEMORY_FAILED; } param_value->set_tensor_addr(tensor_data); diff --git a/mindspore/lite/tools/converter/legacy_optimizer/graph/infershape_pass.cc b/mindspore/lite/tools/converter/legacy_optimizer/graph/infershape_pass.cc index 08f750009f..eff624d9e5 100644 --- a/mindspore/lite/tools/converter/legacy_optimizer/graph/infershape_pass.cc +++ b/mindspore/lite/tools/converter/legacy_optimizer/graph/infershape_pass.cc @@ -32,9 +32,11 @@ namespace { void FreeTensors(std::vector input_tensors, std::vector output_tensors) { for (auto &tensor : input_tensors) { delete tensor; + tensor = nullptr; } for (auto &tensor : output_tensors) { delete tensor; + tensor = nullptr; } input_tensors.clear(); input_tensors.shrink_to_fit();