Home
last modified time | relevance | path

Searched refs:mutable_decision_tree (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
Ddecision-tree-resource.h44 decision_trees::Model* mutable_decision_tree() { in mutable_decision_tree() function
53 return decision_tree_->mutable_decision_tree()->mutable_nodes(id); in get_mutable_tree_node()
Ddecision-tree-resource.cc55 DecisionTree* tree = decision_tree_->mutable_decision_tree(); in SplitNode()
84 DecisionTree* tree = decision_tree_->mutable_decision_tree(); in MaybeInitialize()
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dmodel_ops.cc52 if (!ParseProtoUnlimited(result->mutable_decision_tree(), in Compute()
117 decision_tree_resource->mutable_decision_tree(); in Compute()
Dstats_ops.cc482 auto* node = tree_resource->mutable_decision_tree() in Compute()
483 ->mutable_decision_tree() in Compute()