Searched refs:bucketized_features_list (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | prediction_ops.cc | 40 const OpInputList bucketized_features_list, in ConvertVectorsToMatrices() argument
42 for (const Tensor& tensor : bucketized_features_list) { in ConvertVectorsToMatrices()
74 OpInputList bucketized_features_list; in Compute() local
76 &bucketized_features_list)); in Compute()
78 bucketized_features.reserve(bucketized_features_list.size()); in Compute()
79 ConvertVectorsToMatrices(bucketized_features_list, bucketized_features); in Compute()
217 OpInputList bucketized_features_list; in Compute() local
219 &bucketized_features_list)); in Compute()
221 bucketized_features.reserve(bucketized_features_list.size()); in Compute()
222 ConvertVectorsToMatrices(bucketized_features_list, bucketized_features); in Compute()
[all …]
|
| D | stats_ops.cc | 1265 OpInputList bucketized_features_list; in Compute() local
1267 &bucketized_features_list)); in Compute()
1282 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/core/ops/compat/ops_history_v1/ |
| D | BoostedTreesMakeStatsSummary.pbtxt | 16 name: "bucketized_features_list"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | BoostedTreesMakeStatsSummary.pbtxt | 16 name: "bucketized_features_list"
|
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | stats_ops_test.py | 1336 bucketized_features_list=[[0, 1, 0, 1]],
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 593 …argspec: "args=[\'node_ids\', \'gradients\', \'hessians\', \'bucketized_features_list\', \'max_spl…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 593 …argspec: "args=[\'node_ids\', \'gradients\', \'hessians\', \'bucketized_features_list\', \'max_spl…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 3458 …e, node_ids tf.Output, gradients tf.Output, hessians tf.Output, bucketized_features_list []tf.Outp…
3466 node_ids, gradients, hessians, tf.OutputList(bucketized_features_list),
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 5932 name: "bucketized_features_list"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 11870 name: "bucketized_features_list"
|