| @@ -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); | ||||