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()111 const Eigen::half* logits, const float* sum_probs, in GenerateNormalizedProb()216 Tensor sum_probs; in Compute() local