Searched defs:sum_probs (Results 1 – 1 of 1) sorted by relevance
72 __global__ void GenerateNormalizedProb(const T* logits, const U* sum_probs, in GenerateNormalizedProb()150 Tensor sum_probs; in Compute() local