Searched refs:multinomial_categorical_impl (Results 1 – 1 of 1) sorted by relevance
357 return multinomial_categorical_impl(logits, num_samples, output_dtype, seed)385 return multinomial_categorical_impl(logits, num_samples, dtype, seed)388 def multinomial_categorical_impl(logits, num_samples, dtype, seed): function