Home
last modified time | relevance | path

Searched refs:cached_tree_ids (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/
Dprediction_ops_test.py42 cached_tree_ids = [0, 0]
53 cached_tree_ids=cached_tree_ids,
61 self.assertAllClose(cached_tree_ids, new_tree_ids)
112 cached_tree_ids = [0, 0]
120 cached_tree_ids=cached_tree_ids,
182 cached_tree_ids = [0, 0]
193 cached_tree_ids=cached_tree_ids,
201 self.assertAllClose(cached_tree_ids, new_tree_ids)
293 cached_tree_ids = [0, 0]
303 cached_tree_ids=cached_tree_ids,
[all …]
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dprediction_ops.cc78 const auto cached_tree_ids = cached_tree_ids_t->vec<int32>(); in Compute() local
109 output_tree_ids = cached_tree_ids; in Compute()
114 auto do_work = [&resource, &batch_bucketized_features, &cached_tree_ids, in Compute()
118 int32 tree_id = cached_tree_ids(i); in Compute()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_BoostedTreesTrainingPredict.pbtxt5 name: "cached_tree_ids"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.raw_ops.pbtxt501 …argspec: "args=[\'tree_ensemble_handle\', \'cached_tree_ids\', \'cached_node_ids\', \'bucketized_f…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.raw_ops.pbtxt501 …argspec: "args=[\'tree_ensemble_handle\', \'cached_tree_ids\', \'cached_node_ids\', \'bucketized_f…
/external/tensorflow/tensorflow/go/op/
Dwrappers.go3474 func BoostedTreesTrainingPredict(scope *Scope, tree_ensemble_handle tf.Output, cached_tree_ids tf.O…
3482 tree_ensemble_handle, cached_tree_ids, cached_node_ids, tf.OutputList(bucketized_features),
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt4829 name: "cached_tree_ids"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v1.pbtxt12043 name: "cached_tree_ids"
Dops_history.v2.pbtxt12043 name: "cached_tree_ids"