Searched defs:grad_softmax_row_ptr (Results 1 – 1 of 1) sorted by relevance
560 const T* softmax, const int* grad_softmax_row_ptr, in CSRSparseMatrixSoftmaxGradKernel2D()581 const int* grad_softmax_row_ptr, const int* grad_softmax_col_ind, in CSRSparseMatrixSoftmaxGradKernel3D()630 auto grad_softmax_row_ptr = grad_softmax.row_pointers().vec<int32>(); in CSRSparseMatrixSoftmaxGradGPUImpl() local