Searched refs:GenerateNormalizedProb (Results 1 – 1 of 1) sorted by relevance
72 __global__ void GenerateNormalizedProb(const T* logits, const U* sum_probs, in GenerateNormalizedProb() function184 GenerateNormalizedProb<T, acc_type>, numBlocks, numThreads, 0, in Compute()