Home
last modified time | relevance | path

Searched refs:tree_ensemble_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dresources.cc28 : tree_ensemble_( in BoostedTreesEnsembleResource()
33 return strings::StrCat("TreeEnsemble[size=", tree_ensemble_->trees_size(), in DebugString()
40 if (ParseProtoUnlimited(tree_ensemble_, serialized)) { in InitFromSerialized()
48 return tree_ensemble_->SerializeAsString(); in SerializeAsString()
52 return tree_ensemble_->trees_size(); in num_trees()
58 DCHECK_LT(tree_id, tree_ensemble_->trees_size()); in next_node()
59 DCHECK_LT(node_id, tree_ensemble_->trees(tree_id).nodes_size()); in next_node()
60 const auto& node = tree_ensemble_->trees(tree_id).nodes(node_id); in next_node()
87 DCHECK_LT(tree_id, tree_ensemble_->trees_size()); in node_value()
88 DCHECK_LT(node_id, tree_ensemble_->trees(tree_id).nodes_size()); in node_value()
[all …]
Dresources.h177 boosted_trees::TreeEnsemble* tree_ensemble_; variable