Searched refs:stats_dims (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | stats_ops.cc | 981 const int32 stats_dims = stats_summary_shape(3); in Compute() local
1033 bucket_id, std::vector<float>(stats_dims))); in Compute()
1343 const int64 stats_dims = logits_dims + hessians_dims; in Compute() local
1352 {max_splits_, feature_dims, num_buckets_ + 1, stats_dims}, in Compute()
1367 for (int stat_dim = logits_dims; stat_dim < stats_dims; ++stat_dim) { in Compute()
1467 const int32 stats_dims, in AddInstanceStatsToMap() argument
1476 key, std::vector<float>(stats_dims, 0.0f))); in AddInstanceStatsToMap()
1481 for (int stat_dim = logits_dims; stat_dim < stats_dims; ++stat_dim) { in AddInstanceStatsToMap()
1497 const int32 logits_dims, const int32 stats_dims) { in AddRangeStats() argument
1508 AddInstanceStatsToMap(instance, f_dim, bucket_id, logits_dims, stats_dims, in AddRangeStats()
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_BoostedTreesCalculateBestFeatureSplit.pbtxt | 13 A Rank 4 tensor (#shape=[max_splits, feature_dims, bucket, stats_dims]) for accumulated stats summa…
|
| D | api_def_BoostedTreesCalculateBestFeatureSplitV2.pbtxt | 13 A list of Rank 4 tensor (#shape=[max_splits, feature_dims, bucket, stats_dims]) for accumulated sta…
|
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | stats_ops_test.py | 1524 stats_dims = logits_dims + hess_dims
1535 (max_splits, feature_dims, num_buckets + 1, stats_dims))
|