Home
last modified time | relevance | path

Searched refs:tree_id (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dresources.cc54 const int32 tree_id, const int32 node_id, const int32 index_in_batch, in next_node() argument
56 DCHECK_LT(tree_id, tree_ensemble_->trees_size()); in next_node()
57 DCHECK_LT(node_id, tree_ensemble_->trees(tree_id).nodes_size()); in next_node()
58 const auto& node = tree_ensemble_->trees(tree_id).nodes(node_id); in next_node()
82 const int32 tree_id, const int32 node_id) const { in node_value() argument
83 DCHECK_LT(tree_id, tree_ensemble_->trees_size()); in node_value()
84 DCHECK_LT(node_id, tree_ensemble_->trees(tree_id).nodes_size()); in node_value()
85 const auto& node = tree_ensemble_->trees(tree_id).nodes(node_id); in node_value()
116 void BoostedTreesEnsembleResource::set_node_value(const int32 tree_id, in set_node_value() argument
119 DCHECK_LT(tree_id, tree_ensemble_->trees_size()); in set_node_value()
[all …]
Dresources.h68 const int32 tree_id, const int32 node_id, const int32 index_in_batch,
71 std::vector<float> node_value(const int32 tree_id, const int32 node_id) const;
73 void set_node_value(const int32 tree_id, const int32 node_id,
76 int32 GetNumLayersGrown(const int32 tree_id) const;
78 void SetNumLayersGrown(const int32 tree_id, int32 new_num_layers) const;
86 int64 GetNumNodes(const int32 tree_id);
92 bool is_leaf(const int32 tree_id, const int32 node_id) const;
94 int32 feature_id(const int32 tree_id, const int32 node_id) const;
96 int32 bucket_threshold(const int32 tree_id, const int32 node_id) const;
98 int32 left_id(const int32 tree_id, const int32 node_id) const;
[all …]
Dprediction_ops.cc118 int32 tree_id = cached_tree_ids(i); in Compute() local
126 resource->GetPostPruneCorrection(tree_id, node_id, &node_id, in Compute()
131 const auto& node_logits = resource->node_value(tree_id, node_id); in Compute()
140 if (resource->is_leaf(tree_id, node_id)) { in Compute()
141 const auto& leaf_logits = resource->node_value(tree_id, node_id); in Compute()
147 resource->GetTreeWeight(tree_id) * partial_tree_logit; in Compute()
150 if (tree_id == latest_tree) { in Compute()
154 ++tree_id; in Compute()
157 node_id = resource->next_node(tree_id, node_id, i, in Compute()
232 int32 tree_id = 0; in Compute() local
[all …]
/external/bcc/scripts/
Dgit-clang-format350 tree_id = run('git', 'write-tree')
351 return tree_id
/external/clang/tools/clang-format/
Dgit-clang-format374 tree_id = run('git', 'write-tree')
375 return tree_id
/external/strace/tests-m32/
Dbtrfs.c888 btrfs_print_objectid(key->tree_id); in btrfs_print_search_key()
943 .tree_id = 5, in btrfs_test_search_ioctls()
1044 key_reference.tree_id = 5; in btrfs_test_search_ioctls()
/external/strace/tests-mx32/
Dbtrfs.c888 btrfs_print_objectid(key->tree_id); in btrfs_print_search_key()
943 .tree_id = 5, in btrfs_test_search_ioctls()
1044 key_reference.tree_id = 5; in btrfs_test_search_ioctls()
/external/strace/tests/
Dbtrfs.c888 btrfs_print_objectid(key->tree_id); in btrfs_print_search_key()
943 .tree_id = 5, in btrfs_test_search_ioctls()
1044 key_reference.tree_id = 5; in btrfs_test_search_ioctls()
/external/kernel-headers/original/uapi/linux/
Dbtrfs.h450 __u64 tree_id; /* in */ member
/external/strace/
Dbtrfs.c383 btrfs_print_objectid("{key={", *key, tree_id); in btrfs_print_tree_search()