Searched refs:sample_count_ (Results 1 – 2 of 2) sorted by relevance
50 sample_count_ = getScalarData<int>(*GetInput(operation, operands, kSampleCountParam)); in Multinomial()107 int sample_count_aligned = (sample_count_ + 3) / 4 * 4; in EvalFloat32()133 auto* output_ptr_batch = GetBuffer<int32_t>(output_) + b * sample_count_; in EvalFloat32()134 for (uint64_t j = 0; j < sample_count_; ++j) { in EvalFloat32()
52 int sample_count_; variable