Searched refs:MultinomialOpModel (Results 1 – 1 of 1) sorted by relevance
42 class MultinomialOpModel { class44 MultinomialOpModel(uint32_t batch_size, uint32_t class_size, uint32_t sample_size) in MultinomialOpModel() function in android::nn::wrapper::MultinomialOpModel116 MultinomialOpModel multinomial(kBatchSize, kNumClasses, kNumSamples); in TEST()