Home
last modified time | relevance | path

Searched defs:logits_dimension (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dresources.cc258 const int32_t logits_dimension) { in AddNewTree()
265 const int32_t logits_dimension) { in AddNewTreeWithLogits()
284 const int32_t logits_dimension, int32* left_node_id, int32* right_node_id) { in AddBucketizedSplitNode()
309 const int32_t logits_dimension, int32* left_node_id, int32* right_node_id) { in AddCategoricalSplitNode()
324 const int32_t logits_dimension, int32* left_node_id, int32* right_node_id) { in AddLeafNodes()
381 const int32_t current_tree, const int32_t logits_dimension) { in PostPruneTree()
517 const int logits_dimension = nodes_changes[start_node_id].second.size(); in CalculateParentAndLogitUpdate() local
Dtraining_ops.cc81 int32_t logits_dimension = 1; in Compute() local
/external/tensorflow/tensorflow/core/ops/
Dboosted_trees_ops.cc135 int logits_dimension; in __anon16fb5b110302() local
167 int logits_dimension; in __anon16fb5b110402() local
253 int logits_dimension; in __anon16fb5b110502() local
475 int logits_dimension; in __anon16fb5b110c02() local
553 int logits_dimension; in __anon16fb5b110f02() local
637 int logits_dimension; in __anon16fb5b111102() local