Searched defs:feature_dims (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | stats_ops.cc | 255 const int32 feature_dims = stats_summary_t->dim_size(1); 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()
581 const int32 feature_dims = stats_summaries_list[0].dim_size(1); 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()
1344 const int64 feature_dims = feature_t->dim_size(1); in Compute() local
1496 const int32 feature_dims, const int32 bucket_id, in AddRangeStats()
1565 const int32 feature_dims = feature_shape(1); in Compute() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | stats_ops_test.py | 1521 batch_size, feature_dims, argument
|