diff --git a/mindspore/lite/tools/anf_exporter/anf_exporter.cc b/mindspore/lite/tools/anf_exporter/anf_exporter.cc index 3e3fe2dc2c..7bc8e4fc8a 100644 --- a/mindspore/lite/tools/anf_exporter/anf_exporter.cc +++ b/mindspore/lite/tools/anf_exporter/anf_exporter.cc @@ -594,7 +594,12 @@ int AnfExporter::ConvertInputValueNode(const std::shared_ptr &input_ano paramTensor->dims = {static_cast(shape.size())}; paramTensor->nodeType = schema::NodeType_ValueNode; paramTensor->data.resize(shape.size() * sizeof(int)); - memcpy(paramTensor->data.data(), shape.data(), shape.size() * sizeof(int)); + auto ret = memcpy_s(paramTensor->data.data(), shape.size() * sizeof(int32_t), shape.data(), + shape.size() * sizeof(int32_t)); + if (ret != RET_OK) { + MS_LOG(ERROR) << "memcpy_s data into paramTensor 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));