Browse Source

fix weight quant may print error

pull/13985/head
xutianchun 4 years ago
parent
commit
1307f203c5
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/lite/tools/converter/quantizer/weight_quantizer.cc

+ 1
- 1
mindspore/lite/tools/converter/quantizer/weight_quantizer.cc 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;


Loading…
Cancel
Save