Searched refs:CSRSparseMatrixSoftmaxKernel3D (Results 1 – 1 of 1) sorted by relevance
408 __global__ void CSRSparseMatrixSoftmaxKernel3D( in CSRSparseMatrixSoftmaxKernel3D() function468 TF_CHECK_OK(GpuLaunchKernel(CSRSparseMatrixSoftmaxKernel3D<T>, in CSRSparseMatrixSoftmaxGPUImpl()