Home
last modified time | relevance | path

Searched refs:probability_sum (Results 1 – 2 of 2) sorted by relevance

/frameworks/ml/nn/common/operations/
DMultinomialTest.cpp123 float probability_sum = 0; in TEST() local
126 probability_sum += expf(input[batch_index + i]); in TEST()
131 float probability_expected = expf(input[batch_index + i]) / probability_sum; in TEST()
/frameworks/ml/nn/tools/test_generator/include/
DTestHarness.h409 float probability_sum = 0; in expectMultinomialDistributionWithinTolerance() local
412 probability_sum += expf(input[batch_index + i]); in expectMultinomialDistributionWithinTolerance()
417 float probability_expected = expf(input[batch_index + i]) / probability_sum; in expectMultinomialDistributionWithinTolerance()