Searched defs:n_dist (Results 1 – 2 of 2) sorted by relevance
34 int64_t n_dist = self.dim() > 1 ? self.size(-2) : 1; in multinomial_with_replacement_apply() local138 int64_t n_dist = self.dim() > 1 ? self.size(-2) : 1; in multinomial_with_replacement_apply() local
577 const int64_t n_dist = self.size(0); in multinomial_out() local