Not calculating false branch

bili2002 · Jan 12, 2024 · 0c06b54 · 0c06b54
1 parent 1bac527
commit 0c06b54
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/onnxruntime/core/providers/cpu/ml/tree_ensemble_common.h b/onnxruntime/core/providers/cpu/ml/tree_ensemble_common.h
@@ -673,11 +673,11 @@ void TreeEnsembleCommon<InputType, ThresholdType, OutputType>::ComputeAgg(concur
   if (has_missing_tracks_) {                                                                           \
     while (root->is_not_leaf()) {                                                                      \
       val = x_data[root->feature_id];                                                                  \
-      size_t true_node = root->truenode_or_weight.ptr - &nodes_[0]; \
-      size_t false_node = (root + 1) - &nodes_[0]; \
+      size_t true_node = root->truenode_or_weight.ptr - root; \
+      size_t false_node = 1; \
       bool cond = (val CMP root->value_or_unique_weight || (root->is_missing_track_true() && _isnan_(val))); \
       size_t root_node = true_node * cond + false_node * !cond; \
-      root = &nodes_[root_node];\
+      root = root_node + root;\
     }\
   } else {                                                                                             \
     while (root->is_not_leaf()) {                                                                      \