From 966f1f9fb0e653cfc985ca836ff9da22dea4d0df Mon Sep 17 00:00:00 2001 From: guohongzilong <2713219276@qq.com> Date: Wed, 19 Aug 2020 10:14:52 +0800 Subject: [PATCH] fix quant param nullptr --- .../tools/anf_importer/import_from_meta_graphT.cc | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) 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 cc3b1b6662..39e6a3ab1e 100644 --- a/mindspore/lite/tools/anf_importer/import_from_meta_graphT.cc +++ b/mindspore/lite/tools/anf_importer/import_from_meta_graphT.cc @@ -87,12 +87,16 @@ ValueNodePtr AnfImporterFromMetaGraphT::ConvertPrimitive(const std::unique_ptrquantType == schema::QuantType_AwareTraining) { primTValue->SetQuantType(cNode->quantType); for (int index : cNode->inputIndex) { - std::vector quant_params = {*(meta_graph_->allTensors[index]->quantParams[0])}; - primTValue->AddInputQuantParam(quant_params); + if (meta_graph_->allTensors[index]->quantParams.size() > 0) { + std::vector quant_params = {*(meta_graph_->allTensors[index]->quantParams[0])}; + primTValue->AddInputQuantParam(quant_params); + } } for (int index : cNode->outputIndex) { - std::vector quant_params = {*(meta_graph_->allTensors[index]->quantParams[0])}; - primTValue->AddOutputQuantParam(quant_params); + if (meta_graph_->allTensors[index]->quantParams.size() > 0) { + std::vector quant_params = {*(meta_graph_->allTensors[index]->quantParams[0])}; + primTValue->AddOutputQuantParam(quant_params); + } } } auto value_node = NewValueNode(primTValue);