| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | prediction_ops.cc | 41 std::vector<tensorflow::TTypes<int32>::ConstMatrix>& bucketized_features) { in ConvertVectorsToMatrices() argument
45 bucketized_features.emplace_back( in ConvertVectorsToMatrices()
48 bucketized_features.emplace_back(tensor.matrix<int32>()); in ConvertVectorsToMatrices()
77 std::vector<tensorflow::TTypes<int32>::ConstMatrix> bucketized_features; in Compute() local
78 bucketized_features.reserve(bucketized_features_list.size()); in Compute()
79 ConvertVectorsToMatrices(bucketized_features_list, bucketized_features); in Compute()
80 const int batch_size = bucketized_features[0].dimension(0); in Compute()
121 auto do_work = [&resource, &bucketized_features, &cached_tree_ids, in Compute()
171 resource->next_node(tree_id, node_id, i, bucketized_features); in Compute()
220 std::vector<tensorflow::TTypes<int32>::ConstMatrix> bucketized_features; in Compute() local
[all …]
|
| D | resources.cc | 57 const std::vector<TTypes<int32>::ConstMatrix>& bucketized_features) const { in next_node()
65 const auto bucketized_feature = bucketized_features[split.feature_id()]; in next_node()
73 const auto bucketized_feature = bucketized_features[split.feature_id()]; in next_node()
|
| D | resources.h | 71 const std::vector<TTypes<int32>::ConstMatrix>& bucketized_features) const;
|
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | prediction_ops_test.py | 57 bucketized_features=[feature_0_values, feature_1_values],
94 bucketized_features=[feature_0_values, feature_1_values],
162 bucketized_features=[feature_0_values],
231 bucketized_features=[feature_0_values],
303 bucketized_features=[feature_0_values],
377 bucketized_features=[feature_0_values, feature_1_values],
450 bucketized_features=[feature_0_values, feature_1_values],
542 bucketized_features=[feature_0_values, feature_1_values],
652 bucketized_features=[feature_0_values, feature_1_values],
803 bucketized_features=[feature_0_values, feature_1_values],
[all …]
|
| D | stats_ops_test.py | 1369 bucketized_features = [[3, 1, 2, 0, 1, 2, 0, 1]]
1371 node_ids, gradients, hessians, bucketized_features, max_splits,
1390 bucketized_features = [[3], [1], [2], [0], [1], [2], [0], [1]]
1392 node_ids, gradients, hessians, bucketized_features, max_splits,
1415 bucketized_features = [[3], [1], [2], [0], [missing_feature], [2], [0], [1]]
1417 node_ids, gradients, hessians, bucketized_features, max_splits,
1441 bucketized_features = [[3, 1, 2, 0, 1, 2, 0, 1], [0, 0, 0, 2, 2, 3, 3, 2]]
1443 node_ids, gradients, hessians, bucketized_features, max_splits,
1480 bucketized_features = [[3, 0], [1, 0], [2, 0], [0, 2], [1, 2], [2, 3],
1483 node_ids, gradients, hessians, bucketized_features, max_splits,
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | BoostedTreesPredict.pbtxt | 8 name: "bucketized_features"
|
| D | BoostedTreesExampleDebugOutputs.pbtxt | 8 name: "bucketized_features"
|
| D | BoostedTreesTrainingPredict.pbtxt | 16 name: "bucketized_features"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | BoostedTreesPredict.pbtxt | 8 name: "bucketized_features"
|
| D | BoostedTreesExampleDebugOutputs.pbtxt | 8 name: "bucketized_features"
|
| D | BoostedTreesTrainingPredict.pbtxt | 16 name: "bucketized_features"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BoostedTreesPredict.pbtxt | 5 name: "bucketized_features"
|
| D | api_def_BoostedTreesExampleDebugOutputs.pbtxt | 5 name: "bucketized_features"
|
| D | api_def_BoostedTreesTrainingPredict.pbtxt | 19 name: "bucketized_features"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 577 …argspec: "args=[\'tree_ensemble_handle\', \'bucketized_features\', \'logits_dimension\', \'name\']…
597 …argspec: "args=[\'tree_ensemble_handle\', \'bucketized_features\', \'logits_dimension\', \'name\']…
633 …ree_ensemble_handle\', \'cached_tree_ids\', \'cached_node_ids\', \'bucketized_features\', \'logits…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 577 …argspec: "args=[\'tree_ensemble_handle\', \'bucketized_features\', \'logits_dimension\', \'name\']…
597 …argspec: "args=[\'tree_ensemble_handle\', \'bucketized_features\', \'logits_dimension\', \'name\']…
633 …ree_ensemble_handle\', \'cached_tree_ids\', \'cached_node_ids\', \'bucketized_features\', \'logits…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 3399 …e tf.Output, cached_tree_ids tf.Output, cached_node_ids tf.Output, bucketized_features []tf.Output…
3407 tree_ensemble_handle, cached_tree_ids, cached_node_ids, tf.OutputList(bucketized_features),
13594 func BoostedTreesPredict(scope *Scope, tree_ensemble_handle tf.Output, bucketized_features []tf.Out…
13602 tree_ensemble_handle, tf.OutputList(bucketized_features),
31903 …sExampleDebugOutputs(scope *Scope, tree_ensemble_handle tf.Output, bucketized_features []tf.Output…
31911 tree_ensemble_handle, tf.OutputList(bucketized_features),
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 5825 name: "bucketized_features"
5966 name: "bucketized_features"
6245 name: "bucketized_features"
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 11781 name: "bucketized_features"
11904 name: "bucketized_features"
12051 name: "bucketized_features"
|