Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dresources.cc257 const int32 logits_dimension) { in AddNewTree()
264 const int32 logits_dimension) { in AddNewTreeWithLogits()
283 const int32 logits_dimension, int32* left_node_id, int32* right_node_id) { in AddBucketizedSplitNode()
308 const int32 logits_dimension, int32* left_node_id, int32* right_node_id) { in AddCategoricalSplitNode()
323 const int32 logits_dimension, int32* left_node_id, int32* right_node_id) { in AddLeafNodes()
380 const int32 logits_dimension) { in PostPruneTree()
516 const int logits_dimension = nodes_changes[start_node_id].second.size(); in CalculateParentAndLogitUpdate() local
Dtraining_ops.cc81 int32 logits_dimension = 1; in Compute() local
/external/tensorflow/tensorflow/core/ops/
Dboosted_trees_ops.cc135 int logits_dimension; in __anon8a3fd5e50302() local
167 int logits_dimension; in __anon8a3fd5e50402() local
253 int logits_dimension; in __anon8a3fd5e50502() local
475 int logits_dimension; in __anon8a3fd5e50c02() local
553 int logits_dimension; in __anon8a3fd5e50f02() local
637 int logits_dimension; in __anon8a3fd5e51102() local
Dtensor_forest_ops.cc74 int logits_dimension; in __anon75a11fe00302() local