Home
last modified time | relevance | path

Searched refs:kNumClasses (Results 1 – 2 of 2) sorted by relevance

/frameworks/ml/nn/common/operations/
DMultinomialTest.cpp107 constexpr int kNumClasses = 10000; in TEST() local
111 MultinomialOpModel multinomial(kBatchSize, kNumClasses, kNumSamples); in TEST()
116 class_counts.resize(kNumClasses); in TEST()
125 for (int i = 0; i < kNumClasses; ++i) { in TEST()
128 for (int i = 0; i < kNumClasses; ++i) { in TEST()
/frameworks/ml/nn/tools/test_generator/include/
DTestHarness.h390 const int kNumClasses = 1024; in expectMultinomialDistributionWithinTolerance() local
395 class_counts.resize(kNumClasses); in expectMultinomialDistributionWithinTolerance()
411 for (int i = 0; i < kNumClasses; ++i) { in expectMultinomialDistributionWithinTolerance()
414 for (int i = 0; i < kNumClasses; ++i) { in expectMultinomialDistributionWithinTolerance()