Searched refs:bucketized_features_list (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | prediction_ops.cc | 65 OpInputList bucketized_features_list; in Compute() local
67 &bucketized_features_list)); in Compute()
69 batch_bucketized_features.reserve(bucketized_features_list.size()); in Compute()
70 for (const Tensor& tensor : bucketized_features_list) { in Compute()
204 OpInputList bucketized_features_list; in Compute() local
206 &bucketized_features_list)); in Compute()
208 batch_bucketized_features.reserve(bucketized_features_list.size()); in Compute()
209 for (const Tensor& tensor : bucketized_features_list) { in Compute()
308 OpInputList bucketized_features_list; in Compute() local
310 &bucketized_features_list)); in Compute()
[all …]
|
| D | stats_ops.cc | 231 OpInputList bucketized_features_list; in Compute() local
233 &bucketized_features_list)); in Compute()
248 const auto& features = bucketized_features_list[feature_idx].vec<int32>(); in Compute()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BoostedTreesMakeStatsSummary.pbtxt | 23 name: "bucketized_features_list"
49 int; inferred from the size of bucketized_features_list; the number of features.
|
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | stats_ops_test.py | 339 bucketized_features_list=[[0, 1, 0, 1]],
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 469 …argspec: "args=[\'node_ids\', \'gradients\', \'hessians\', \'bucketized_features_list\', \'max_spl…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 469 …argspec: "args=[\'node_ids\', \'gradients\', \'hessians\', \'bucketized_features_list\', \'max_spl…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 3553 …e, node_ids tf.Output, gradients tf.Output, hessians tf.Output, bucketized_features_list []tf.Outp…
3561 node_ids, gradients, hessians, tf.OutputList(bucketized_features_list),
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 4656 name: "bucketized_features_list"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 11870 name: "bucketized_features_list"
|
| D | ops_history.v2.pbtxt | 11870 name: "bucketized_features_list"
|