Searched refs:CalculateWeightsAndGains (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
D | stats_ops.cc | 121 CalculateWeightsAndGains(total_grad, total_hess, l1, l2, &unused, in Compute() 130 CalculateWeightsAndGains(cum_grad_bucket, cum_hess_bucket, l1, l2, in Compute() 135 CalculateWeightsAndGains(total_grad - cum_grad_bucket, in Compute()
|
D | tree_helper.h | 32 static void CalculateWeightsAndGains(const float g, const float h, in CalculateWeightsAndGains() function
|
D | training_ops.cc | 282 CalculateWeightsAndGains(gradients_mean, hessians_mean, l1, l2, &logits, in Compute()
|