Home
last modified time | relevance | path

Searched defs:hessian_dim (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dstats_ops.cc259 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/
Dboosted_trees_ops.cc384 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