Searched refs:class_size (Results 1 – 2 of 2) sorted by relevance
100 const int class_size = SizeOfDimension(input_, 1); in EvalFloat32() local115 const float* input_ptr_batch = inputData + b * class_size; in EvalFloat32()117 for (uint64_t j = 0; j < class_size; ++j) { in EvalFloat32()125 cdf.resize(class_size); in EvalFloat32()126 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()