Browse Source

!13985 fix weight quant may print error log

From: @xutianchun
Reviewed-by: @zhang_xue_tong,@zhanghaibo5
Signed-off-by: @zhanghaibo5
pull/13985/MERGE
mindspore-ci-bot Gitee 5 years ago
parent
commit
69dfcb3936
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