Searched refs:stat_dim (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | stats_ops.cc | 1027 const int32 stat_dim = stats_summary_indices(idx, 3); in Compute() local
1035 stats[stat_dim] = stats_summary_values(idx); in Compute()
1363 for (int stat_dim = 0; stat_dim < logits_dims; ++stat_dim) { in Compute() local
1364 temp_stats_double(node, feature_dim, bucket, stat_dim) += in Compute()
1365 gradients(i, stat_dim); in Compute()
1367 for (int stat_dim = logits_dims; stat_dim < stats_dims; ++stat_dim) { in Compute() local
1368 temp_stats_double(node, feature_dim, bucket, stat_dim) += in Compute()
1369 hessians(i, stat_dim - logits_dims); in Compute()
1478 for (int stat_dim = 0; stat_dim < logits_dims; ++stat_dim) { in AddInstanceStatsToMap() local
1479 stats[stat_dim] += gradients(instance, stat_dim); in AddInstanceStatsToMap()
[all …]
|