Home
last modified time | relevance | path

Searched defs:MultinomialOpModel (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DMultinomialTest.cpp42 class MultinomialOpModel { class
44 MultinomialOpModel(uint32_t batch_size, uint32_t class_size, uint32_t sample_size) in MultinomialOpModel() function in android::nn::wrapper::MultinomialOpModel
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dmultinomial_test.cc55 class MultinomialOpModel : public tflite::SingleOpModel { class
57 MultinomialOpModel(tflite::TensorData logits, int num_samples, in MultinomialOpModel() function in tflite::__anon752fffe90111::MultinomialOpModel
H A Drandom_ops_test.cc63 class MultinomialOpModel : public SingleOpModel { class
65 MultinomialOpModel(InputType input_type, in MultinomialOpModel() function in tflite::__anon9905f3300111::MultinomialOpModel