Searched refs:class_size (Results 1 – 2 of 2) sorted by relevance
107 const int class_size = SizeOfDimension(input_, 1); in EvalFloat32() local122 const float* input_ptr_batch = inputData + b * class_size; in EvalFloat32()124 for (uint64_t j = 0; j < class_size; ++j) { in EvalFloat32()132 cdf.resize(class_size); in EvalFloat32()133 for (uint64_t j = 0; j < class_size; ++j) { in EvalFloat32()
39 MultinomialOpModel(uint32_t batch_size, uint32_t class_size, uint32_t sample_size) in MultinomialOpModel() argument40 : batch_size_(batch_size), class_size_(class_size), sample_size_(sample_size) { in MultinomialOpModel()