Searched refs:max_logits_ (Results 1 – 1 of 1) sorted by relevance
101 : logits_(logits), max_logits_(max_logits), num_cols_(num_cols) {} in SubtractAndExpFunctor()106 strict_cast<U>(logits_[gid] - ldg(max_logits_ + gid / num_cols_)); in operator ()()111 const T* max_logits_; member