Searched refs:multinomial_categorical_impl (Results 1 – 1 of 1) sorted by relevance
496 return multinomial_categorical_impl(logits, num_samples, output_dtype, seed)525 return multinomial_categorical_impl(logits, num_samples, dtype, seed)528 def multinomial_categorical_impl(logits, num_samples, dtype, seed): function