提交 7e0a6816 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!4753 fix quant param null

Merge pull request !4753 from ghzl/fix-quant-param-null
......@@ -87,12 +87,16 @@ ValueNodePtr AnfImporterFromMetaGraphT::ConvertPrimitive(const std::unique_ptr<s
if (cNode->quantType == schema::QuantType_AwareTraining) {
primTValue->SetQuantType(cNode->quantType);
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) {
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);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册