Searched refs:kNumClasses (Results 1 – 3 of 3) sorted by relevance
107 constexpr int kNumClasses = 10000; in TEST() local111 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()
391 const int kNumClasses = 1024; in expectMultinomialDistributionWithinTolerance() local396 class_counts.resize(kNumClasses); in expectMultinomialDistributionWithinTolerance()412 for (int i = 0; i < kNumClasses; ++i) { in expectMultinomialDistributionWithinTolerance()415 for (int i = 0; i < kNumClasses; ++i) { in expectMultinomialDistributionWithinTolerance()
154 const int kNumClasses = inputOperand.dimensions[1]; in expectMultinomialDistributionWithinTolerance() local159 std::vector<int> classCounts(kNumClasses); in expectMultinomialDistributionWithinTolerance()180 for (int i = 0; i < kNumClasses; ++i) { in expectMultinomialDistributionWithinTolerance()183 for (int i = 0; i < kNumClasses; ++i) { in expectMultinomialDistributionWithinTolerance()