Searched refs:class_size_ (Results 1 – 1 of 1) sorted by relevance
45 : batch_size_(batch_size), class_size_(class_size), sample_size_(sample_size) { in MultinomialOpModel()47 OperandType logitsType(Type::TENSOR_FLOAT32, {batch_size_, class_size_}); in MultinomialOpModel()72 const int sample_count = batch_size_ * class_size_; in Invoke()103 const uint32_t class_size_; member in android::nn::wrapper::MultinomialOpModel