Fix bug

bili2002 · Jan 19, 2024 · dbcbeaf · dbcbeaf
1 parent a9e3d41
commit dbcbeaf
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/onnxruntime/core/providers/cpu/ml/tree_ensemble_common.h b/onnxruntime/core/providers/cpu/ml/tree_ensemble_common.h
@@ -295,7 +295,9 @@ Status TreeEnsembleCommon<InputType, ThresholdType, OutputType>::Init(
     return nodes_[left.second].feature_id < nodes_[right.second].feature_id;
   });
 
-  updated_mapping.fill(0);
+  for (auto& curr : updated_mapping) {
+    curr = 0;
+  }
   nodes_.clear();
   nodes_.reserve(limit);