Searched refs:CSRSparseMatrixSoftmaxKernel2D (Results 1 – 1 of 1) sorted by relevance
370 __global__ void CSRSparseMatrixSoftmaxKernel2D(const int rows, in CSRSparseMatrixSoftmaxKernel2D() function434 TF_CHECK_OK(GpuLaunchKernel(CSRSparseMatrixSoftmaxKernel2D<T>, in CSRSparseMatrixSoftmaxGPUImpl()