Searched refs:batch_index (Results 1 – 2 of 2) sorted by relevance
124 const int batch_index = kBatchSize * b; in TEST() local126 probability_sum += expf(input[batch_index + i]); in TEST()131 float probability_expected = expf(input[batch_index + i]) / probability_sum; in TEST()
410 const int batch_index = kBatchSize * b; in expectMultinomialDistributionWithinTolerance() local412 probability_sum += expf(input[batch_index + i]); in expectMultinomialDistributionWithinTolerance()417 float probability_expected = expf(input[batch_index + i]) / probability_sum; in expectMultinomialDistributionWithinTolerance()