Searched refs:softmax_col (Results 1 – 1 of 1) sorted by relevance
511 const int softmax_col = ldg(softmax_col_ind + i); in CalculateRowSoftmaxGrad() local513 if (softmax_col == grad_softmax_col) { in CalculateRowSoftmaxGrad()517 } else if (softmax_col > grad_softmax_col) { in CalculateRowSoftmaxGrad()532 const int softmax_col = ldg(softmax_col_ind + i); in CalculateRowSoftmaxGrad() local540 if (softmax_col == grad_softmax_col) { in CalculateRowSoftmaxGrad()544 } else if (softmax_col > grad_softmax_col) { in CalculateRowSoftmaxGrad()