Searched refs:CSRSparseMatrixSoftmaxGPUImpl (Results 1 – 1 of 1) sorted by relevance
432 Status CSRSparseMatrixSoftmaxGPUImpl(OpKernelContext* ctx, in CSRSparseMatrixSoftmaxGPUImpl() function483 return CSRSparseMatrixSoftmaxGPUImpl<T>(ctx, logits, softmax_values); \