Searched refs:curr_hess (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | stats_ops.cc | 507 Eigen::VectorXf curr_hess = stats_vec.tail(hessian_dim); in CalculateBestEqualitySplit() local
508 MaybeUpdateBestSplit(curr_grad, total_grad - curr_grad, curr_hess, in CalculateBestEqualitySplit()
509 total_hess - curr_hess, logits_dim, bucket, f_dim, in CalculateBestEqualitySplit()
891 Eigen::VectorXf curr_hess = stats_vec.tail(hessian_dim); in CalculateBestEqualitySplit() local
892 MaybeUpdateBestSplit(curr_grad, total_grad - curr_grad, curr_hess, in CalculateBestEqualitySplit()
893 total_hess - curr_hess, logits_dim, bucket, f_dim, in CalculateBestEqualitySplit()
|