Searched defs:stats_dims (Results 1 – 1 of 1) sorted by relevance
981 const int32 stats_dims = stats_summary_shape(3); in Compute() local1343 const int64 stats_dims = logits_dims + hessians_dims; in Compute() local1467 const int32 stats_dims, in AddInstanceStatsToMap()1497 const int32 logits_dims, const int32 stats_dims) { in AddRangeStats()1563 const int64 stats_dims = logits_dims + hessians_dims; in Compute() local