Searched defs:logits_dims (Results 1 – 2 of 2) sorted by relevance
1341 const int64 logits_dims = gradients_t->dim_size(1); in Compute() local1466 const int32 logits_dims, in AddInstanceStatsToMap()1497 const int32 logits_dims, const int32 stats_dims) { in AddRangeStats()1561 const int64 logits_dims = gradients_t->dim_size(1); in Compute() local
1522 logits_dims, hess_dims): argument