Searched refs:SparseTensorToCOOMatrixKernel (Results 1 – 1 of 1) sorted by relevance
149 __global__ void SparseTensorToCOOMatrixKernel(const int64* indices, in SparseTensorToCOOMatrixKernel() function170 TF_CHECK_OK(GpuLaunchKernel(SparseTensorToCOOMatrixKernel<2>, in operator ()()175 TF_CHECK_OK(GpuLaunchKernel(SparseTensorToCOOMatrixKernel<3>, in operator ()()