Searched refs:kBatchSize (Results 1 – 3 of 3) 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()
390 const int kBatchSize = 1; in expectMultinomialDistributionWithinTolerance() local409 for (int b = 0; b < kBatchSize; ++b) { in expectMultinomialDistributionWithinTolerance()411 const int batch_index = kBatchSize * b; in expectMultinomialDistributionWithinTolerance()
153 const int kBatchSize = inputOperand.dimensions[0]; in expectMultinomialDistributionWithinTolerance() local177 for (int b = 0; b < kBatchSize; ++b) { in expectMultinomialDistributionWithinTolerance()179 const int batchIndex = kBatchSize * b; in expectMultinomialDistributionWithinTolerance()