Searched refs:sum_prod (Results 1 – 1 of 1) sorted by relevance
508 T sum_prod = 0; in CalculateRowSoftmaxGrad() local514 sum_prod += ldg(softmax + i) * ldg(grad_softmax + j); in CalculateRowSoftmaxGrad()541 gradient[i] = (ldg(grad_softmax + j) - sum_prod) * ldg(softmax + i); in CalculateRowSoftmaxGrad()551 gradient[i] = -sum_prod * ldg(softmax + i); in CalculateRowSoftmaxGrad()