From 3f68f575c411d49742139ba314da023a7e003937 Mon Sep 17 00:00:00 2001 From: xutianchun Date: Mon, 17 Aug 2020 10:57:44 +0800 Subject: [PATCH] fix post training quant --- mindspore/lite/tools/anf_exporter/anf_exporter.cc | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/mindspore/lite/tools/anf_exporter/anf_exporter.cc b/mindspore/lite/tools/anf_exporter/anf_exporter.cc index a704d6b8f3..b4e4c6a1bc 100644 --- a/mindspore/lite/tools/anf_exporter/anf_exporter.cc +++ b/mindspore/lite/tools/anf_exporter/anf_exporter.cc @@ -238,7 +238,6 @@ schema::MetaGraphT *AnfExporter::Export(const FuncGraphPtr &func_graph) { } meta_graphT->nodes.emplace_back(std::move(node)); - primitiveT_value->SetPrimitiveT(nullptr); } // set graph input tensors SetGraphInputIndex(meta_graphT); @@ -296,6 +295,18 @@ int AnfExporter::ConvertInputParameter(const std::shared_ptr input_anod paramTensor->nodeType = schema::NodeType_ValueNode; paramTensor->data.resize(paramValue->tensor_size()); memcpy(paramTensor->data.data(), paramValue->tensor_addr(), paramValue->tensor_size()); + for (auto &ite : paramValue->quant_param()) { + auto quantPar = std::make_unique(); + quantPar->scale = ite->scale; + quantPar->zeroPoint = ite->zeroPoint; + quantPar->min = ite->zeroPoint; + quantPar->max = ite->max; + quantPar->narrowRange = ite->narrowRange; + quantPar->inited = ite->inited; + quantPar->numBits = ite->numBits; + paramTensor->quantParams.emplace_back(std::move(quantPar)); + paramTensor->dataType = paramValue->tensor_type(); + } } node_id_map_[paramNode->fullname_with_scope()] = meta_graphT->allTensors.size(); output_cnode->inputIndex.emplace_back(meta_graphT->allTensors.size());