Home
last modified time | relevance | path

Searched refs:cum_hess (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dstats_ops.cc110 std::vector<float> cum_hess; in Compute() local
112 cum_hess.reserve(num_buckets); in Compute()
122 cum_hess.clear(); in Compute()
130 cum_hess.push_back(total_hess); in Compute()
149 const float cum_hess_bucket = cum_hess[bucket]; in Compute()
431 std::vector<Eigen::VectorXf> cum_hess; in CalculateBestInequalitySplit() local
434 cum_hess.reserve(num_buckets); in CalculateBestInequalitySplit()
443 cum_hess.clear(); in CalculateBestInequalitySplit()
458 cum_hess.push_back(total_hess); in CalculateBestInequalitySplit()
474 cum_hess[bucket] + missing_bucket_hess; in CalculateBestInequalitySplit()
[all …]