Searched defs:hessian_dim (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | stats_ops.cc | 259 const int32 hessian_dim = stats_summary_t->dim_size(3) - logits_dim; in Compute() local 425 const int32 feature_dims, const int32 logits_dim, const int32 hessian_dim, in CalculateBestInequalitySplit() 495 const int32 hessian_dim, const int32 num_buckets, const float l1, in CalculateBestEqualitySplit() 585 const int32 hessian_dim = stats_summaries_list[0].dim_size(3) - logits_dim; in Compute() local 809 const int32 feature_dims, const int32 logits_dim, const int32 hessian_dim, in CalculateBestInequalitySplit() 879 const int32 hessian_dim, const int32 num_buckets, const float l1, in CalculateBestEqualitySplit()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | boosted_trees_ops.cc | 384 DimensionHandle hessian_dim = c->Dim(c->input(2), 1); in __anon770f674c0a02() local 443 DimensionHandle hessian_dim = c->Dim(c->input(2), 1); in __anon770f674c0b02() local
|