Searched defs:logits_dims (Results 1 – 2 of 2) sorted by relevance
1391 const int64_t logits_dims = gradients_t->dim_size(1); in Compute() local1516 const int32_t logits_dims, const int32_t stats_dims, in AddInstanceStatsToMap()1545 const int32_t logits_dims, const int32_t stats_dims) { in AddRangeStats()1610 const int64_t logits_dims = gradients_t->dim_size(1); in Compute() local
1522 logits_dims, hess_dims): argument