Searched refs:MultinomialOpModel (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | random_ops_test.cc | 63 class MultinomialOpModel : public SingleOpModel { class
65 MultinomialOpModel(InputType input_type, in MultinomialOpModel() function in tflite::__anon9905f3300111::MultinomialOpModel
257 MultinomialOpModel m1(GetParam(), kLogits, kNumBatches, kNumClasses, in TEST_P()
267 MultinomialOpModel m2(GetParam(), kLogits, kNumBatches, kNumClasses, in TEST_P()
287 MultinomialOpModel m1(GetParam(), kLogits, kNumBatches, kNumClasses, in TEST_P()
298 MultinomialOpModel m2(GetParam(), kLogits, kNumBatches, kNumClasses, in TEST_P()
330 MultinomialOpModel m(/*input_type=*/InputType::kConst, kLogits, kNumBatches, in TEST()
356 MultinomialOpModel m(/*input_type=*/InputType::kConst, kLogits, kNumBatches, in TEST()
386 MultinomialOpModel m(/*input_type=*/InputType::kConst, kLogits, kNumBatches, in TEST()
416 MultinomialOpModel m(/*input_type=*/InputType::kConst, kLogits, kNumBatches, in TEST()
[all …]
|
| H A D | multinomial_test.cc | 55 class MultinomialOpModel : public tflite::SingleOpModel { class
57 MultinomialOpModel(tflite::TensorData logits, int num_samples, in MultinomialOpModel() function in tflite::__anon752fffe90111::MultinomialOpModel
106 tflite::MultinomialOpModel m({tflite::GetTTEnum<Float>(), {3, 3}}, in TYPED_TEST()
136 tflite::MultinomialOpModel m({tflite::GetTTEnum<Float>(), {1, 3}}, in TYPED_TEST()
160 tflite::MultinomialOpModel m({tflite::GetTTEnum<Float>(), {1, 3}}, in TYPED_TEST()
185 tflite::MultinomialOpModel m({tflite::GetTTEnum<Float>(), {1, kNumLogits}}, in TYPED_TEST()
211 tflite::MultinomialOpModel m({tflite::GetTTEnum<Float>(), {1, kNumLogits}}, in TYPED_TEST()
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | MultinomialTest.cpp | 42 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
116 MultinomialOpModel multinomial(kBatchSize, kNumClasses, kNumSamples); in TEST()
|