Searched defs:logits_dim (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | stats_ops.cc | 258 const int32 logits_dim = logits_dim_; in Compute() local
425 const int32 feature_dims, const int32 logits_dim, const int32 hessian_dim, in CalculateBestInequalitySplit()
494 const int32 node_id, const int32 feature_dims, const int32 logits_dim, in CalculateBestEqualitySplit()
521 const int32 logits_dim, const int32 bucket, in MaybeUpdateBestSplit()
584 const int32 logits_dim = logits_dim_; in Compute() local
809 const int32 feature_dims, const int32 logits_dim, const int32 hessian_dim, in CalculateBestInequalitySplit()
878 const int32 node_id, const int32 feature_dims, const int32 logits_dim, in CalculateBestEqualitySplit()
905 const int32 logits_dim, const int32 bucket, in MaybeUpdateBestSplit()
|
| D | tree_helper.h | 70 const int32 logits_dim = g.size(); in CalculateWeightsAndGains() local
|
| D | training_ops.cc | 475 const int32 logits_dim = mean_gradients_t->dim_size(1); in Compute() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | boosted_trees_ops.cc | 383 DimensionHandle logits_dim = c->Dim(c->input(1), 1); in __anon770f674c0a02() local
442 DimensionHandle logits_dim = c->Dim(c->input(1), 1); in __anon770f674c0b02() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | stats_ops_test.py | 1676 logits_dim = 2 variable in BestMultiDimFeatureSplitMultiClassV2Op
|