Searched defs:softmax_row_ptr (Results 1 – 1 of 1) sorted by relevance
546 const int rows, const int* softmax_row_ptr, const int* softmax_col_ind, in CSRSparseMatrixSoftmaxGradKernel2D()567 const int* softmax_row_ptr, const int* softmax_col_ind, const T* softmax, in CSRSparseMatrixSoftmaxGradKernel3D()613 auto softmax_row_ptr = softmax.row_pointers().vec<int32>(); in CSRSparseMatrixSoftmaxGradGPUImpl() local