Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dstats_ops.cc100 float total_grad = 0.0; in Compute() local
104 total_grad += stats_summary[feature_idx](node_id, bucket, 0); in Compute()
106 cum_grad.push_back(total_grad); in Compute()
121 CalculateWeightsAndGains(total_grad, total_hess, l1, l2, &unused, in Compute()
135 CalculateWeightsAndGains(total_grad - cum_grad_bucket, in Compute()