Searched defs:tree_logits (Results 1 – 2 of 2) sorted by relevance
51 def _dnn_to_tree_squared_loss(dnn_logits, tree_logits, example_weights): argument63 def _dnn_to_tree_cross_entropy_loss(dnn_logits, tree_logits, example_weights): argument
231 std::vector<float> tree_logits(logits_dimension_, 0.0); in Compute() local341 const auto& tree_logits = resource->node_value(0, 0); in Compute() local366 const auto& tree_logits = resource->node_value(tree_id, node_id); in Compute() local