Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dstats_ops.cc506 Eigen::VectorXf curr_grad = stats_vec.head(logits_dim); in CalculateBestEqualitySplit() local
508 MaybeUpdateBestSplit(curr_grad, total_grad - curr_grad, curr_hess, in CalculateBestEqualitySplit()
890 Eigen::VectorXf curr_grad = stats_vec.head(logits_dim); in CalculateBestEqualitySplit() local
892 MaybeUpdateBestSplit(curr_grad, total_grad - curr_grad, curr_hess, in CalculateBestEqualitySplit()