Searched refs:batch_size_ (Results 1 – 1 of 1) sorted by relevance
40 : batch_size_(batch_size), class_size_(class_size), sample_size_(sample_size) { in MultinomialOpModel()42 OperandType logitsType(Type::TENSOR_FLOAT32, {batch_size_, class_size_}); in MultinomialOpModel()50 OperandType outputType(Type::TENSOR_INT32, {batch_size_, sample_size_}); in MultinomialOpModel()67 const int sample_count = batch_size_ * class_size_; in Invoke()83 output_.insert(output_.end(), batch_size_ * sample_size_, 0); in Invoke()97 const uint32_t batch_size_; member in android::nn::wrapper::MultinomialOpModel