Searched defs:softmax_col_ind (Results 1 – 1 of 1) sorted by relevance
493 const int softmax_begin, const int softmax_end, const int* softmax_col_ind, in CalculateRowSoftmaxGrad()559 const int rows, const int* softmax_row_ptr, const int* softmax_col_ind, in CSRSparseMatrixSoftmaxGradKernel2D()580 const int* softmax_row_ptr, const int* softmax_col_ind, const T* softmax, in CSRSparseMatrixSoftmaxGradKernel3D()627 auto softmax_col_ind = softmax.col_indices().vec<int32>(); in CSRSparseMatrixSoftmaxGradGPUImpl() local