Searched defs:grad_softmax_col_ind (Results 1 – 1 of 1) sorted by relevance
495 const int* grad_softmax_col_ind, const T* grad_softmax, T* gradient) { in CalculateRowSoftmaxGrad()561 const int* grad_softmax_col_ind, const T* grad_softmax, T* gradient) { in CSRSparseMatrixSoftmaxGradKernel2D()581 const int* grad_softmax_row_ptr, const int* grad_softmax_col_ind, in CSRSparseMatrixSoftmaxGradKernel3D()631 auto grad_softmax_col_ind = grad_softmax.col_indices().vec<int32>(); in CSRSparseMatrixSoftmaxGradGPUImpl() local