Browse Source

modify static check

pull/8956/head
yvette 5 years ago
parent
commit
1a520c4d79
4 changed files with 10 additions and 1 deletions
  1. +1
    -0
      mindspore/lite/src/runtime/kernel/arm/base/pooling_base.cc
  2. +5
    -1
      mindspore/lite/tools/anf_exporter/anf_exporter.cc
  3. +1
    -0
      mindspore/lite/tools/anf_importer/import_from_meta_graphT.cc
  4. +3
    -0
      mindspore/lite/tools/converter/legacy_optimizer/graph/infershape_pass.cc

+ 1
- 0
mindspore/lite/src/runtime/kernel/arm/base/pooling_base.cc View File

@@ -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);


+ 5
- 1
mindspore/lite/tools/anf_exporter/anf_exporter.cc View File

@@ -471,7 +471,11 @@ int AnfExporter::ConvertInputValueNode(const std::shared_ptr<AnfNode> &input_ano
paramTensor->dims = {static_cast<int32_t>(shape.size())};
paramTensor->nodeType = schema::NodeType_ValueNode;
paramTensor->data.resize(shape.size() * sizeof(int));
memcpy(paramTensor->data.data(), shape.data(), shape.size() * sizeof(int));
if (memcpy_s(paramTensor->data.data(), shape.size() * sizeof(int), shape.data(), shape.size() * sizeof(int)) !=
RET_OK) {
MS_LOG(ERROR) << "memcpy_s failed";
return RET_ERROR;
}
node_id_map_[valueNode->fullname_with_scope()] = meta_graphT->allTensors.size();
output_cnode->inputIndex.emplace_back(meta_graphT->allTensors.size());
meta_graphT->allTensors.emplace_back(std::move(paramTensor));


+ 1
- 0
mindspore/lite/tools/anf_importer/import_from_meta_graphT.cc View File

@@ -58,6 +58,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);


+ 3
- 0
mindspore/lite/tools/converter/legacy_optimizer/graph/infershape_pass.cc View File

@@ -32,15 +32,18 @@ namespace {
void FreeTensors(std::vector<Tensor *> input_tensors, std::vector<Tensor *> 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();
output_tensors.clear();
output_tensors.shrink_to_fit();
}

std::vector<Tensor *> ConvertTensorToLiteTensor(MetaGraphT *graph, const std::vector<uint32_t> &tensor_indexs,
const schema::PrimitiveType node_type) {
MS_ASSERT(graph != nullptr);


Loading…
Cancel
Save