Searched refs:multinomial (Results 1 – 2 of 2) sorted by relevance
111 MultinomialOpModel multinomial(kBatchSize, kNumClasses, kNumSamples); in TEST() local112 multinomial.Invoke(); in TEST()114 std::vector<uint32_t> output = multinomial.GetOutput(); in TEST()121 std::vector<float> input = multinomial.GetInput(); in TEST()
1140 Multinomial multinomial(operation, mOperands); in executeOperation() local1144 multinomial.Eval(); in executeOperation()