Searched refs:kBatchSize (Results 1 – 2 of 2) sorted by relevance
106 constexpr int kBatchSize = 8; in TEST() local111 MultinomialOpModel multinomial(kBatchSize, kNumClasses, kNumSamples); in TEST()122 for (int b = 0; b < kBatchSize; ++b) { in TEST()124 const int batch_index = kBatchSize * b; in TEST()
389 const int kBatchSize = 1; in expectMultinomialDistributionWithinTolerance() local408 for (int b = 0; b < kBatchSize; ++b) { in expectMultinomialDistributionWithinTolerance()410 const int batch_index = kBatchSize * b; in expectMultinomialDistributionWithinTolerance()