Searched refs:hessians_dims (Results 1 – 1 of 1) sorted by relevance
1342 const int64 hessians_dims = hessians_t->dim_size(1); in Compute() local1343 const int64 stats_dims = logits_dims + hessians_dims; in Compute()1562 const int64 hessians_dims = hessians_t->dim_size(1); in Compute() local1563 const int64 stats_dims = logits_dims + hessians_dims; in Compute()