Searched refs:class_size (Results 1 – 2 of 2) sorted by relevance
113 const uint32_t class_size = SizeOfDimension(input_, 1); in EvalFloat32() local128 const float* input_ptr_batch = inputData + b * class_size; in EvalFloat32()130 for (uint64_t j = 0; j < class_size; ++j) { in EvalFloat32()138 cdf.resize(class_size); in EvalFloat32()139 for (uint64_t j = 0; j < class_size; ++j) { in EvalFloat32()
44 MultinomialOpModel(uint32_t batch_size, uint32_t class_size, uint32_t sample_size) in MultinomialOpModel() argument45 : batch_size_(batch_size), class_size_(class_size), sample_size_(sample_size) { in MultinomialOpModel()