Searched refs:batch_index (Results 1 – 2 of 2) sorted by relevance
129 const int batch_index = kBatchSize * b; in TEST() local131 probability_sum += expf(input[batch_index + i]); in TEST()136 float probability_expected = expf(input[batch_index + i]) / probability_sum; in TEST()
411 const int batch_index = kBatchSize * b; in expectMultinomialDistributionWithinTolerance() local413 probability_sum += expf(input[batch_index + i]); in expectMultinomialDistributionWithinTolerance()418 float probability_expected = expf(input[batch_index + i]) / probability_sum; in expectMultinomialDistributionWithinTolerance()