Searched refs:decision_tree (Results 1 – 13 of 13) sorted by relevance
37 decision_tree_->decision_tree().nodes_size(), "]"); in DebugString()42 const decision_trees::Model& decision_tree() const { return *decision_tree_; } in decision_tree() function49 return decision_tree_->decision_tree().nodes(id).leaf(); in get_leaf()
32 const DecisionTree& tree = decision_tree_->decision_tree(); in TraverseTree()
13 PROTO_INPUT(tensorflow::decision_trees::DecisionTree, decision_tree);14 PROTO_IN_OUT(tensorflow::decision_trees::DecisionTree, decision_tree);
21 DecisionTree decision_tree = 1; field
88 decision_tree_resource->decision_tree().SerializeAsString(); in Compute()143 decision_tree_resource->decision_tree().decision_tree().nodes_size(); in Compute()393 const auto& tree = decision_tree_resource->decision_tree(); in Compute()404 for (const auto& node : tree.decision_tree().nodes()) { in Compute()
388 tree_resource->decision_tree().decision_tree().nodes_size() < in Compute()480 int num_nodes = tree_resource->decision_tree().decision_tree().nodes_size(); in Compute()
51 const boosted_trees::Tree& decision_tree() const;
22 const boosted_trees::Tree& TensorForestTreeResource::decision_tree() const { in decision_tree() function in tensorflow::TensorForestTreeResource
67 decision_tree_resource->decision_tree().SerializeAsString(); in Compute()
234 srcs = ["trees/decision_tree.cc"],235 hdrs = ["trees/decision_tree.h"],
71 // to create the decision_tree::Leaf and the second is used to infer
134 tree = member.submodel.decision_tree
59 "${tensorflow_source_dir}/tensorflow/contrib/boosted_trees/lib/trees/decision_tree.cc"