Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dtree_helper.h70 const int32 logits_dim = g.size(); in CalculateWeightsAndGains() local
71 if (logits_dim == 1) { in CalculateWeightsAndGains()
101 } else if (h.size() == logits_dim * logits_dim) { /* Full Hessian */ in CalculateWeightsAndGains()
103 identity.setIdentity(logits_dim, logits_dim); in CalculateWeightsAndGains()
106 h.reshaped(logits_dim, logits_dim) + l2 * identity; in CalculateWeightsAndGains()
108 } else if (h.size() == logits_dim) { /* Diagonal Hessian approximation. */ in CalculateWeightsAndGains()
122 Eigen::ArrayXf ones = Eigen::ArrayXf::Ones(logits_dim); in CalculateWeightsAndGains()
Dstats_ops.cc258 const int32 logits_dim = logits_dim_; in Compute() local
259 const int32 hessian_dim = stats_summary_t->dim_size(3) - logits_dim; in Compute()
261 DCHECK_LE(hessian_dim, logits_dim * logits_dim); in Compute()
302 Eigen::VectorXf best_contrib_for_left(logits_dim); in Compute()
303 Eigen::VectorXf best_contrib_for_right(logits_dim); in Compute()
308 num_buckets + 1, logits_dim + hessian_dim); in Compute()
310 stats_mat.leftCols(logits_dim).colwise().sum(); in Compute()
316 Eigen::VectorXf parent_weight(logits_dim); in Compute()
322 stats_summary, node_id, feature_dims, logits_dim, hessian_dim, in Compute()
329 logits_dim, hessian_dim, num_buckets, l1, l2, &best_gain, in Compute()
[all …]
Dtraining_ops.cc475 const int32 logits_dim = mean_gradients_t->dim_size(1); in Compute() local
492 DCHECK_EQ(logits_dim, 1); in Compute()
/external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/
Dstats_ops_test.py1676 logits_dim = 2 variable in BestMultiDimFeatureSplitMultiClassV2Op
1727 hessians = nodes_agg[:, self.logits_dim:]
1838 logits_dimension=self.logits_dim))
1851 logits_dimension=self.logits_dim))
1882 logits_dimension=self.logits_dim))
1914 logits_dimension=self.logits_dim))
1948 logits_dimension=self.logits_dim))
1981 logits_dimension=self.logits_dim))
2014 logits_dimension=self.logits_dim))
2049 logits_dimension=self.logits_dim))
[all …]
/external/tensorflow/tensorflow/core/ops/
Dboosted_trees_ops.cc383 DimensionHandle logits_dim = c->Dim(c->input(1), 1); in __anon770f674c0a02() local
387 TF_RETURN_IF_ERROR(c->Add(logits_dim, hessian_dim, &stats_dim)); in __anon770f674c0a02()
442 DimensionHandle logits_dim = c->Dim(c->input(1), 1); in __anon770f674c0b02() local
445 TF_RETURN_IF_ERROR(c->Add(logits_dim, hessian_dim, &stats_dim)); in __anon770f674c0b02()