Searched refs:CuSparseSpGEMMDescriptor (Results 1 – 2 of 2) sorted by relevance
276 class TORCH_CUDA_CPP_API CuSparseSpGEMMDescriptor279 CuSparseSpGEMMDescriptor() { in CuSparseSpGEMMDescriptor() function
720 auto spgemm_desc = at::cuda::sparse::CuSparseSpGEMMDescriptor(); in spgemm()