Browse Source

fix quant param nullptr

tags/v0.7.0-beta
guohongzilong 5 years ago
parent
commit
966f1f9fb0
1 changed files with 8 additions and 4 deletions
  1. +8
    -4
      mindspore/lite/tools/anf_importer/import_from_meta_graphT.cc

+ 8
- 4
mindspore/lite/tools/anf_importer/import_from_meta_graphT.cc View File

@@ -87,12 +87,16 @@ ValueNodePtr AnfImporterFromMetaGraphT::ConvertPrimitive(const std::unique_ptr<s
if (cNode->quantType == schema::QuantType_AwareTraining) { if (cNode->quantType == schema::QuantType_AwareTraining) {
primTValue->SetQuantType(cNode->quantType); primTValue->SetQuantType(cNode->quantType);
for (int index : cNode->inputIndex) { for (int index : cNode->inputIndex) {
std::vector<schema::QuantParamT> quant_params = {*(meta_graph_->allTensors[index]->quantParams[0])};
primTValue->AddInputQuantParam(quant_params);
if (meta_graph_->allTensors[index]->quantParams.size() > 0) {
std::vector<schema::QuantParamT> quant_params = {*(meta_graph_->allTensors[index]->quantParams[0])};
primTValue->AddInputQuantParam(quant_params);
}
} }
for (int index : cNode->outputIndex) { for (int index : cNode->outputIndex) {
std::vector<schema::QuantParamT> quant_params = {*(meta_graph_->allTensors[index]->quantParams[0])};
primTValue->AddOutputQuantParam(quant_params);
if (meta_graph_->allTensors[index]->quantParams.size() > 0) {
std::vector<schema::QuantParamT> quant_params = {*(meta_graph_->allTensors[index]->quantParams[0])};
primTValue->AddOutputQuantParam(quant_params);
}
} }
} }
auto value_node = NewValueNode(primTValue); auto value_node = NewValueNode(primTValue);


Loading…
Cancel
Save