Searched refs:CuSparseConstDnMatDescriptor (Results 1 – 3 of 3) sorted by relevance
137 class TORCH_CUDA_CPP_API CuSparseConstDnMatDescriptor140 explicit CuSparseConstDnMatDescriptor(const Tensor& input, int64_t batch_offset = -1);169 class TORCH_CUDA_CPP_API CuSparseConstDnMatDescriptor174 explicit CuSparseConstDnMatDescriptor(
119 CuSparseConstDnMatDescriptor::CuSparseConstDnMatDescriptor(const Tensor& input, int64_t batch_offse… in CuSparseConstDnMatDescriptor() function in at::cuda::sparse::CuSparseConstDnMatDescriptor
635 auto descB = at::cuda::sparse::CuSparseConstDnMatDescriptor( in spmm()1449 auto descA = at::cuda::sparse::CuSparseConstDnMatDescriptor(*A_, /*batch_offset=*/i); in sampled_addmm_out_sparse_csr()1450 auto descB = at::cuda::sparse::CuSparseConstDnMatDescriptor(*B_, /*batch_offset=*/i); in sampled_addmm_out_sparse_csr()