Searched refs:sum_probs (Results 1 – 1 of 1) sorted by relevance
72 __global__ void GenerateNormalizedProb(const T* logits, const U* sum_probs, in GenerateNormalizedProb() argument88 result = input - max_val - log(ldg(sum_probs + row)); in GenerateNormalizedProb()90 result = exp(input - max_val) / ldg(sum_probs + row); in GenerateNormalizedProb()150 Tensor sum_probs; in Compute() local158 softmax_out->shape(), &sum_probs)); in Compute()180 context, const_cast<acc_type*>(sum_probs.flat<acc_type>().data()), in Compute()186 reinterpret_cast<const acc_type*>(sum_probs.flat<acc_type>().data()), in Compute()