Home
last modified time | relevance | path

Searched refs:kNumSamples (Results 1 – 3 of 3) sorted by relevance

/frameworks/ml/nn/common/operations/
DMultinomialTest.cpp108 constexpr int kNumSamples = 128; in TEST() local
111 MultinomialOpModel multinomial(kBatchSize, kNumClasses, kNumSamples); in TEST()
130 static_cast<float>(class_counts[i]) / static_cast<float>(kNumSamples); in TEST()
/frameworks/ml/nn/tools/test_generator/test_harness/include_legacy/
DTestHarness.h392 const int kNumSamples = 128; in expectMultinomialDistributionWithinTolerance() local
417 static_cast<float>(class_counts[i]) / static_cast<float>(kNumSamples); in expectMultinomialDistributionWithinTolerance()
/frameworks/ml/nn/tools/test_generator/test_harness/
DTestHarness.cpp155 const int kNumSamples = outputOperand.dimensions[1]; in expectMultinomialDistributionWithinTolerance() local
185 static_cast<float>(classCounts[i]) / static_cast<float>(kNumSamples); in expectMultinomialDistributionWithinTolerance()