Searched refs:probability_expected (Results 1 – 2 of 2) sorted by relevance
136 float probability_expected = expf(input[batch_index + i]) / probability_sum; in TEST() local137 EXPECT_THAT(probability, FloatNear(probability_expected, kMaxProbabilityDelta)); in TEST()
418 float probability_expected = expf(input[batch_index + i]) / probability_sum; in expectMultinomialDistributionWithinTolerance() local420 ::testing::FloatNear(probability_expected, in expectMultinomialDistributionWithinTolerance()