Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dtree_helper.h58 const Eigen::MatrixXf& hessian_and_reg, const Eigen::VectorXf& g, in MultiDimLogitSolveForWeightAndGain() argument
60 *weight = -hessian_and_reg.colPivHouseholderQr().solve(g); in MultiDimLogitSolveForWeightAndGain()
105 Eigen::MatrixXf hessian_and_reg = in CalculateWeightsAndGains() local
107 MultiDimLogitSolveForWeightAndGain(hessian_and_reg, g, weight, gain); in CalculateWeightsAndGains()
110 Eigen::ArrayXf hessian_and_reg = h.array() + l2; in CalculateWeightsAndGains() local
113 for (int i = 0; i < hessian_and_reg.size(); ++i) { in CalculateWeightsAndGains()
114 if (hessian_and_reg[i] == 0.0) { in CalculateWeightsAndGains()
123 Eigen::ArrayXf temp = ones / hessian_and_reg; in CalculateWeightsAndGains()
129 MultiDimLogitSolveForWeightAndGain(hessian_and_reg.matrix().asDiagonal(), in CalculateWeightsAndGains()