HomeSort by relevance Sort by last modified time
    Searched refs:n_draws (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/python/ops/distributions/
multinomial.py 247 n_draws = math_ops.cast(self.total_count, dtype=dtypes.int32)
251 n_draws = array_ops.ones_like(
252 self.logits[..., 0], dtype=n_draws.dtype) * n_draws
254 n_draws[..., array_ops.newaxis], dtype=self.logits.dtype) * self.logits
258 flat_ndraws = n * array_ops.reshape(n_draws, [-1]) # [B1B2...Bm]
dirichlet_multinomial.py 262 n_draws = math_ops.cast(self.total_count, dtype=dtypes.int32)
273 num_samples=n_draws,

Completed in 116 milliseconds