Searched refs:num_trees (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | resource_ops_test.py | 40 (_, num_trees, num_finalized_trees, num_attempted_layers,
42 self.assertEqual(0, self.evaluate(num_trees))
157 (stamp_token, num_trees, num_finalized_trees, num_attempted_layers,
160 self.assertEqual(2, self.evaluate(num_trees))
171 (stamp_token, num_trees, num_finalized_trees, num_attempted_layers,
174 self.assertEqual(0, self.evaluate(num_trees))
223 (stamp_token, num_trees, num_finalized_trees, num_attempted_layers,
226 self.assertEqual(1, self.evaluate(num_trees))
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | resource_ops.cc | 88 const int num_trees = tree_ensemble_resource->num_trees(); in Compute() local
90 (num_trees <= 0 || in Compute()
91 tree_ensemble_resource->IsTreeFinalized(num_trees - 1)) in Compute()
92 ? num_trees in Compute()
93 : num_trees - 1; in Compute()
118 output_num_trees_t->scalar<int32>()() = num_trees; in Compute()
|
| D | training_ops.cc | 137 if (ensemble_resource->num_trees() > 0) { in Compute()
151 int32 num_trees = resource->num_trees(); in UpdateGlobalAttemptsAndRetrieveGrowableTree() local
152 int32 current_tree = num_trees - 1; in UpdateGlobalAttemptsAndRetrieveGrowableTree()
160 if (num_trees <= 0) { in UpdateGlobalAttemptsAndRetrieveGrowableTree()
357 if (ensemble_resource->num_trees() > 0) { in Compute()
371 int32 num_trees = resource->num_trees(); in UpdateGlobalAttemptsAndRetrieveGrowableTree() local
372 int32 current_tree = num_trees - 1; in UpdateGlobalAttemptsAndRetrieveGrowableTree()
380 if (num_trees <= 0) { in UpdateGlobalAttemptsAndRetrieveGrowableTree()
503 if (ensemble_resource->num_trees() == 0) { in Compute()
|
| D | prediction_ops.cc | 111 const int32 latest_tree = resource->num_trees() - 1; in Compute()
233 if (resource->num_trees() <= 0) { in Compute()
238 const int32 last_tree = resource->num_trees() - 1; in Compute()
336 const int32 last_tree = resource->num_trees() - 1; in Compute()
|
| D | resources.cc | 51 int32 BoostedTreesEnsembleResource::num_trees() const { in num_trees() function in tensorflow::BoostedTreesEnsembleResource
236 DCHECK_LT(tree_id, num_trees()); in SetTreeWeight()
244 const int n_trees = num_trees(); in UpdateGrowingMetadata()
|
| D | resources.h | 59 int32 num_trees() const;
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | BoostedTreesGetEnsembleStates.pbtxt | 12 name: "num_trees"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | BoostedTreesGetEnsembleStates.pbtxt | 12 name: "num_trees"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BoostedTreesGetEnsembleStates.pbtxt | 17 name: "num_trees"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | boosted_trees_ops.py | 271 (stamp_token, num_trees, num_finalized_trees, num_attempted_layers,
277 array_ops.identity(num_trees, name='num_trees'),
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 47979 …s(scope *Scope, tree_ensemble_handle tf.Output) (stamp_token tf.Output, num_trees tf.Output, num_f…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 5874 name: "num_trees"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 11812 name: "num_trees"
|