Searched refs:cached_tree_ids (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | prediction_ops_test.py | 44 cached_tree_ids = [0, 0]
55 cached_tree_ids=cached_tree_ids,
63 self.assertAllClose(cached_tree_ids, new_tree_ids)
78 cached_tree_ids = [0, 0]
92 cached_tree_ids=cached_tree_ids,
100 self.assertAllClose(cached_tree_ids, new_tree_ids)
152 cached_tree_ids = [0, 0]
160 cached_tree_ids=cached_tree_ids,
221 cached_tree_ids = [0, 0]
229 cached_tree_ids=cached_tree_ids,
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | prediction_ops.cc | 85 const auto cached_tree_ids = cached_tree_ids_t->vec<int32>(); in Compute() local
116 output_tree_ids = cached_tree_ids; in Compute()
121 auto do_work = [&resource, &bucketized_features, &cached_tree_ids, in Compute()
126 int32 tree_id = cached_tree_ids(i); in Compute()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | BoostedTreesTrainingPredict.pbtxt | 8 name: "cached_tree_ids"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | BoostedTreesTrainingPredict.pbtxt | 8 name: "cached_tree_ids"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BoostedTreesTrainingPredict.pbtxt | 5 name: "cached_tree_ids"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 633 …argspec: "args=[\'tree_ensemble_handle\', \'cached_tree_ids\', \'cached_node_ids\', \'bucketized_f…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 633 …argspec: "args=[\'tree_ensemble_handle\', \'cached_tree_ids\', \'cached_node_ids\', \'bucketized_f…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 3399 func BoostedTreesTrainingPredict(scope *Scope, tree_ensemble_handle tf.Output, cached_tree_ids tf.O…
3407 tree_ensemble_handle, cached_tree_ids, cached_node_ids, tf.OutputList(bucketized_features),
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 6237 name: "cached_tree_ids"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 12043 name: "cached_tree_ids"
|