Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DMultinomial.cpp120 int sample_count_aligned = (sample_count_ + 3) / 4 * 4; in EvalFloat32() local
122 sample_count_aligned *= 2; in EvalFloat32()
124 random_generator.ReserveRandomOutputs(batch_size * sample_count_aligned, 256); in EvalFloat32()