Searched refs:softmax_end (Results 1 – 1 of 1) sorted by relevance
493 const int softmax_begin, const int softmax_end, const int* softmax_col_ind, in CalculateRowSoftmaxGrad() argument510 i < softmax_end && j < grad_softmax_end;) { in CalculateRowSoftmaxGrad()528 (softmax_begin == softmax_end) in CalculateRowSoftmaxGrad()530 : ldg(softmax_col_ind + softmax_end - 1) + 1; in CalculateRowSoftmaxGrad()531 for (int i = softmax_begin, j = grad_softmax_begin; i < softmax_end;) { in CalculateRowSoftmaxGrad()