Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Ddistillation_loss.py51 def _dnn_to_tree_squared_loss(dnn_logits, tree_logits, example_weights): argument
54 logits=_logits_to_label_for_tree(tree_logits, n_classes),
63 def _dnn_to_tree_cross_entropy_loss(dnn_logits, tree_logits, example_weights): argument
67 logits=tree_logits,
72 logits=tree_logits,
Ddnn_tree_combined_estimator.py270 tree_logits = predictions_dict["predictions"]
293 dnn_logits_fixed, tree_logits, weight_tensor)
306 tree_train_logits = tree_logits
310 lambda: tree_logits,
313 tree_train_logits = dnn_logits + tree_logits
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dprediction_ops.cc231 std::vector<float> tree_logits(logits_dimension_, 0.0); in Compute() local
240 tree_logits[j] += tree_weight * leaf_logits[j]; in Compute()
255 output_logits(i, j) = tree_logits[j]; in Compute()
341 const auto& tree_logits = resource->node_value(0, 0); in Compute() local
342 DCHECK_EQ(tree_logits.size(), 1); in Compute()
343 float tree_logit = resource->GetTreeWeight(0) * tree_logits[0]; in Compute()
366 const auto& tree_logits = resource->node_value(tree_id, node_id); in Compute() local
367 DCHECK_EQ(tree_logits.size(), 1); in Compute()
368 tree_logit = resource->GetTreeWeight(tree_id) * tree_logits[0]; in Compute()