Home
last modified time | relevance | path

Searched refs:multinomial (Results 1 – 3 of 3) sorted by relevance

/frameworks/ml/nn/common/operations/
DMultinomialTest.cpp111 MultinomialOpModel multinomial(kBatchSize, kNumClasses, kNumSamples); in TEST() local
112 multinomial.Invoke(); in TEST()
114 std::vector<uint32_t> output = multinomial.GetOutput(); in TEST()
121 std::vector<float> input = multinomial.GetInput(); in TEST()
/frameworks/ml/nn/common/
DCpuExecutor.cpp1120 Multinomial multinomial(operation, operands); in executeOperation() local
1124 multinomial.Eval(); in executeOperation()
/frameworks/ml/nn/tools/api/
Dtypes.spec4968 * Draws samples from a multinomial distribution.