Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DMultinomial.cpp107 int sample_count_aligned = (sample_count_ + 3) / 4 * 4; in EvalFloat32() local
109 sample_count_aligned *= 2; in EvalFloat32()
111 random_generator.ReserveRandomOutputs(batch_size * sample_count_aligned, 256); in EvalFloat32()