fix weight quant may print error

This commit is contained in:
xutianchun 2021-03-24 17:27:40 +08:00
parent b6bf797ae7
commit 1307f203c5
1 changed files with 1 additions and 1 deletions
mindspore/lite/tools/converter/quantizer

View File

@ -672,7 +672,7 @@ STATUS WeightQuantizer::DoMixedQuant(FuncGraphPtr func_graph) {
STATUS WeightQuantizer::DoFixedQuant(FuncGraphPtr func_graph) {
MS_ASSERT(func_graph != nullptr);
for (auto &cnode : func_graph->GetOrderedCnodes()) {
auto primitive = GetValueNode<PrimitivePtr>(cnode->input(0));
auto primitive = GetValueNode<std::shared_ptr<ops::PrimitiveC>>(cnode->input(0));
if (primitive == nullptr) {
MS_LOG(DEBUG) << cnode->fullname_with_scope() << " : primitive is nullptr";
continue;