Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DMultinomial.cpp63 sample_count_ = getScalarData<int>(*GetInput(operation, operands, kSampleCountParam)); in Multinomial()
120 int sample_count_aligned = (sample_count_ + 3) / 4 * 4; in EvalFloat32()
146 auto* output_ptr_batch = GetBuffer<int32_t>(output_) + b * sample_count_; in EvalFloat32()
147 for (uint64_t j = 0; j < static_cast<uint64_t>(sample_count_); ++j) { in EvalFloat32()
/packages/modules/NeuralNetworks/common/types/operations/include/
DMultinomial.h51 int sample_count_; variable